Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DLstm.cpp217 linear_op_contexts.emplace_back(create_linear_context(w_ii.t(), b_ii)); in pack_lstm_linear_op_contexts()
218 linear_op_contexts.emplace_back(create_linear_context(w_hi.t(), b_hi)); in pack_lstm_linear_op_contexts()
219 linear_op_contexts.emplace_back(create_linear_context(w_if.t(), b_if)); in pack_lstm_linear_op_contexts()
220 linear_op_contexts.emplace_back(create_linear_context(w_hf.t(), b_hf)); in pack_lstm_linear_op_contexts()
221 linear_op_contexts.emplace_back(create_linear_context(w_ig.t(), b_ig)); in pack_lstm_linear_op_contexts()
222 linear_op_contexts.emplace_back(create_linear_context(w_hg.t(), b_hg)); in pack_lstm_linear_op_contexts()
223 linear_op_contexts.emplace_back(create_linear_context(w_io.t(), b_io)); in pack_lstm_linear_op_contexts()
224 linear_op_contexts.emplace_back(create_linear_context(w_ho.t(), b_ho)); in pack_lstm_linear_op_contexts()
H A DGru.cpp177 linear_op_contexts.emplace_back(create_linear_context(w_ir.t(), b_ir)); in pack_linear_op_contexts()
178 linear_op_contexts.emplace_back(create_linear_context(w_hr.t(), b_hr)); in pack_linear_op_contexts()
179 linear_op_contexts.emplace_back(create_linear_context(w_iz.t(), b_iz)); in pack_linear_op_contexts()
180 linear_op_contexts.emplace_back(create_linear_context(w_hz.t(), b_hz)); in pack_linear_op_contexts()
181 linear_op_contexts.emplace_back(create_linear_context(w_in.t(), b_in)); in pack_linear_op_contexts()
182 linear_op_contexts.emplace_back(create_linear_context(w_hn.t(), b_hn)); in pack_linear_op_contexts()
H A DMm.h98 c10::intrusive_ptr<LinearPackedContext> create_linear_context(
H A DRegister.cpp284 TORCH_FN(create_linear_context)); in TORCH_LIBRARY_IMPL()
H A DMm.cpp975 c10::intrusive_ptr<LinearPackedContext> create_linear_context( in create_linear_context() function
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_quantized_api_test.cpp3362 at::native::vulkan::ops::create_linear_context( in _test_quantized_linear()