/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
H A D | Convolution.cpp | 62 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 D | Convolution.cpp | 31 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 D | convert.cc | 98 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 D | op_convolution_test.cpp | 34 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 D | op_convolution_backward_test.cpp | 36 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 D | Convolution.cpp | 34 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 D | Convolution.cpp | 263 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() 1724 … bool 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 D | DFT.cpp | 348 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 D | kernel_ops_util.cpp | 204 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 D | BatchRulesConvolution.cpp | 20 …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 D | test_flop_counter.py | 162 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 D | op_convolution.cpp | 57 bool transposed) { in conv2d_impl() 200 bool transposed, in convolution_wrapper() 354 bool transposed, in convolution_out()
|
H A D | op_convolution_backward.cpp | 34 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 D | quantized_matmul_out.cpp | 67 bool transposed, in _typed_quantized_matmul() 121 bool transposed, in quantized_matmul_out()
|
/aosp_15_r20/external/executorch/backends/vulkan/_passes/ |
H A D | custom_ops_defs.py | 38 transposed=False, argument 80 transposed=False, argument
|
/aosp_15_r20/external/executorch/backends/vulkan/ |
H A D | custom_ops_lib.py | 38 transposed=False, argument 80 transposed=False, argument
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/ |
H A D | Conv.cpp | 27 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 D | maia_extension.cpp | 37 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 D | shape_util_test.cc | 818 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 D | vl_idct.c | 74 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 D | MoveTransposeUpTests.cpp | 19 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 D | fast_hadamard_transform_test_impl.h | 29 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 D | packedtensoraccessor_test.cpp | 21 auto transposed = original.transpose(0, 2); in TEST() local
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | TensorModeKernel.cpp | 58 auto transposed = self.transpose(dim, ndim - 1); in mode_kernel_impl() local
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/ |
H A D | decompose_silu.cpp | 20 bool transposed = Operator::Bool(inputToSilu, 6); in shouldDecomposeSilu() local
|