Searched refs:a_view_ (Results 1 – 2 of 2) sorted by relevance
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/ |
| H A D | gemm_schedule.hpp | 406 const view_t &a_view() const { return a_view_; } in a_view()
411 set_view(v, a_view_); in set_a_view()
412 bmnk_mapper_.set_a_vars(a_view_.vvars()); in set_a_view()
427 return a_view_.create_sub_view(a_tg_tile_); in a_tg_view()
853 for (auto *view : {&a_view_, &b_view_, &c_view_}) { in init_problem_tiles()
860 a_view_.vvars(), split_infos, tile_level_t::thread_group); in init_problem_tiles()
866 a_view_.vvars(), split_infos, tile_level_t::thread); in init_problem_tiles()
966 view_t a_view_; member in dnnl::impl::gpu::jit::gemm_schedule_t
|
| H A D | kernel_builder.cpp | 4254 , a_view_(a_view) in multiply_builder_t()
4294 oss << "A view: " << a_view_ << std::endl; in str()
4319 ir_assert(a_view_.can_convert_to_vlayout()) in try_build_dpas()
4324 auto a_layout = a_view_.create_vlayout(); in try_build_dpas()
4421 for (auto *view : {&a_view_, &b_view_}) { in create_loop_nest()
4503 std::vector<dim_t> tile_dims(a_view_.nvdims(), 1); in try_build_mad_impl()
4505 auto layout = a_view_.create_pseudo_vlayout(); in try_build_mad_impl()
4537 auto &a_type = a_view_.type(); in build_mad()
4554 std::vector<int> a_idx(a_view_.nvdims()); in build_mad()
4565 int a_off = a_view_(a_idx) * a_type.size(); in build_mad()
[all …]
|