Home
last modified time | relevance | path

Searched refs:b_reduce_ctx_ (Results 1 – 1 of 1) sorted by relevance

/dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/
H A Dkernel_builder.cpp4802 , b_reduce_ctx_(b_reduce_ctx) in load_multiply_builder_t()
4821 b_reduce_ctx_.init_reduced_thr_view( in load_multiply_builder_t()
5003 auto reduce_stmt = b_reduce_ctx_.create_reduce_stmt( in load_sub_tile()
5085 b_reduce_context_t &b_reduce_ctx_; member in dnnl::impl::gpu::jit::load_multiply_builder_t
5270 , b_reduce_ctx_(cfg) in compute_builder_t()
5330 b_reduce_ctx_.set_b_reduced_mem_buf(buf); in set_b_reduced_mem_buf()
5334 b_reduce_ctx_.set_b_reduced_view(v); in set_b_reduced_view()
5342 b_reduce_ctx_.set_reduce_condition(cond); in set_reduce_condition()
5406 auto &ctx = b_reduce_ctx_; in build()
5624 auto reduce_stmt = b_reduce_ctx_.create_reduce_stmt( in prepare_gmem_to_slm_impl()
[all …]