Home
last modified time | relevance | path

Searched defs:transposed (Results 1 – 25 of 60) sorted by relevance

123

/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DConvolution.cpp62 const bool transposed, in determine_method()
291 const bool transposed, in get_shader()
377 const bool transposed) { in record_op()
450 const bool transposed) { in record_quantized_op()
513 const bool transposed, in pack_weights()
548 const bool transposed, in pack_biases()
576 const bool transposed) { in compute_overlay_region()
634 const bool transposed, in bias_valid()
663 const bool transposed, in available()
772 const bool transposed, in convolution()
[all …]
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DConvolution.cpp31 const bool transposed = graph->get_bool(extra_args[4]); in resize_conv2d_node() local
90 const bool transposed, in prepack_biases()
243 const bool transposed) { in create_conv2d_params()
254 void check_conv2d_params(const Kernel2dParams& p, const bool transposed) { in check_conv2d_params()
272 const bool transposed) { in get_conv2d_method()
312 const ValueRef transposed, in add_conv2d_node()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dconvert.cc98 std::vector<float> transposed(GetElementsSizeForPHWO4I4(tensor.shape)); in ConvertToPHWO4I4() local
108 std::vector<float> transposed(GetElementsSizeForPHWO4I4(tensor.shape)); in ConvertToPHWO4I4Transposed() local
164 std::vector<float> transposed(GetElementsSizeForPHWO4I4(tensor.shape)); in ConvertToPHWO4I4() local
210 std::vector<float> transposed(GetElementsSizeForPIOHW4(tensor.shape)); in ConvertToPIOHW4() local
369 std::vector<float> transposed(GetElementsSizeForPHWC4(tensor.shape)); in ConvertToPHWC4() local
381 std::vector<float> transposed(GetElementsSizeForPHWC4(batched_shape)); in ConvertToPHWC4() local
/aosp_15_r20/external/executorch/kernels/test/
H A Dop_convolution_test.cpp34 bool transposed, in op_convolution_out()
534 bool transposed = true; in TEST_F() local
571 bool transposed = true; in TEST_F() local
639 bool transposed = true; in TEST_F() local
683 bool transposed = true; in TEST_F() local
713 bool transposed = true; in TEST_F() local
H A Dop_convolution_backward_test.cpp36 bool transposed, in op_convolution_backward_out()
169 bool transposed = false; in TEST_F() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/
H A DConvolution.cpp34 const bool transposed, in available()
96 const bool transposed, in create_and_run()
179 const bool transposed, in create()
478 const bool transposed) { in use_convolution2d()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvolution.cpp263 const bool transposed) { in xnnpack_use_convolution2d()
275 const bool transposed) { in xnnpack_use_convolution2d()
289 bool transposed{}; member
664 bool transposed = params.transposed; in check_shape_forward() local
852 bool transposed, in complex_convolution()
1176 bool transposed, IntArrayRef output_padding, int64_t groups) { in convolution()
1192 bool transposed, IntArrayRef output_padding, int64_t groups) { in convolution_overrideable()
1724bool transposed, IntArrayRef output_padding, int64_t groups, std::array<bool, 3> output_mask) { in convolution_backward_overrideable() argument
1995 …IntArrayRef stride, IntArrayRef padding, IntArrayRef dilation, bool transposed, IntArrayRef output… in convolution_backward()
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DDFT.cpp348 auto transposed = permute(first_pass, PermutationVector(1U, 0U)); in rdft_2d() local
359 auto transposed = permute(src, PermutationVector(1U, 0U)); in ridft_2d() local
377 auto transposed = permute(first_pass, PermutationVector(1U, 0U)); in dft_2d() local
383 auto transposed = permute(src, PermutationVector(1U, 0U)); in dft_2d() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/util/
H A Dkernel_ops_util.cpp204 bool transposed, in _kernel_output_size_helper()
232 bool transposed, in calculate_kernel_output_sizes()
326 bool transposed, in check_convolution_args()
412 bool transposed, in get_convolution_out_target_size()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesConvolution.cpp20 …f stride, c10::SymIntArrayRef padding, c10::SymIntArrayRef dilation, bool transposed, c10::SymIntA… in convolution_batch_rule()
172 …:SymIntArrayRef stride, c10::SymIntArrayRef padding, c10::SymIntArrayRef dilation, bool transposed, in convolution_backward_input_batch_rule()
253 …:SymIntArrayRef stride, c10::SymIntArrayRef padding, c10::SymIntArrayRef dilation, bool transposed, in convolution_backward_weight_batch_rule()
364 …:SymIntArrayRef stride, c10::SymIntArrayRef padding, c10::SymIntArrayRef dilation, bool transposed, in convolution_backward_plumbing()
/aosp_15_r20/external/pytorch/test/
H A Dtest_flop_counter.py162 def forward(inp, weight, transposed): argument
193 def boring_conv(x, weight, transposed): argument
199 def only_convs(x, weight, transposed): argument
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_convolution.cpp57 bool transposed) { in conv2d_impl()
200 bool transposed, in convolution_wrapper()
354 bool transposed, in convolution_out()
H A Dop_convolution_backward.cpp34 bool transposed, in check_convolution_backward_args()
238 bool transposed, in convolution_backward_out()
/aosp_15_r20/external/executorch/backends/cadence/reference/operators/
H A Dquantized_matmul_out.cpp67 bool transposed, in _typed_quantized_matmul()
121 bool transposed, in quantized_matmul_out()
/aosp_15_r20/external/executorch/backends/vulkan/_passes/
H A Dcustom_ops_defs.py38 transposed=False, argument
80 transposed=False, argument
/aosp_15_r20/external/executorch/backends/vulkan/
H A Dcustom_ops_lib.py38 transposed=False, argument
80 transposed=False, argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/
H A DConv.cpp27 bool transposed; member
160 bool transposed = params.transposed; in check_shape_forward() local
605 bool transposed, in convolution_backward_overrideable()
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dmaia_extension.cpp37 bool transposed, IntArrayRef output_padding, int64_t groups) { in fake_convolution()
46 IntArrayRef dilation, bool transposed, IntArrayRef output_padding, in fake_convolution_backward()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_util_test.cc818 Shape transposed = ShapeUtil::MakeShapeWithLayout(F32, {64, 128}, {0, 1}); in TEST() local
824 Shape transposed = ShapeUtil::MakeShapeWithLayout(F32, {128, 64}, {0, 1}); in TEST() local
831 Shape transposed = in TEST() local
840 Shape transposed = in TEST() local
/aosp_15_r20/external/mesa3d/src/gallium/auxiliary/vl/
H A Dvl_idct.c74 bool transposed, float size) in calc_addr()
97 struct ureg_src saddr[2], bool right_side, bool transposed, in increment_addr()
/aosp_15_r20/external/armnn/src/armnn/test/optimizations/
H A DMoveTransposeUpTests.cpp19 const armnn::TensorInfo transposed({ 1, 3, 5, 2 }, armnn::DataType::Float32); variable
/aosp_15_r20/external/executorch/extension/llm/custom_ops/spinquant/test/
H A Dfast_hadamard_transform_test_impl.h29 std::unique_ptr<T[]> transposed = std::make_unique<T[]>(28 * vec_size); in fast_hadamard_transform_28N_with_transpose() local
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dpackedtensoraccessor_test.cpp21 auto transposed = original.transpose(0, 2); in TEST() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DTensorModeKernel.cpp58 auto transposed = self.transpose(dim, ndim - 1); in mode_kernel_impl() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/
H A Ddecompose_silu.cpp20 bool transposed = Operator::Bool(inputToSilu, 6); in shouldDecomposeSilu() local

123