| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_as_strided_copy_test.cpp | 44 const std::vector<int32_t> in_sizes = {3, 3}; in test_detach_copy_out() local
76 const std::vector<int32_t> in_sizes = {3, 3}; in test_as_strided_copy_out_invalid_parameters() local
145 const std::vector<int32_t> in_sizes = {3, 3}; in test_detach_copy_out() local
169 const std::vector<int32_t> in_sizes = {3, 3}; in test_detach_copy_out() local
212 const std::vector<int32_t> in_sizes = {3, 3}; in TEST_F() local
|
| H A D | op_detach_copy_test.cpp | 49 const std::vector<int32_t> in_sizes = {2, 2}; in test_detach_copy_out_invalid_shape() local
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/utils/ |
| H A D | KernelUtils.cpp | 76 const std::vector<int64_t>& in_sizes, in calc_out_sizes_hw()
119 const std::vector<int64_t>& in_sizes, in calc_transpose_out_sizes_hw()
150 const std::vector<int64_t>& in_sizes, in calc_out_sizes_hw()
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | NativeLayerNorm.cpp | 39 std::vector<int64_t> in_sizes = in->sizes(); in resize_native_layer_norm_node() local
89 std::vector<int64_t> in_sizes = t_input->sizes(); in add_native_layer_norm_node() local
|
| H A D | Repeat.cpp | 75 const std::vector<int64_t>& in_sizes = t_in->sizes(); in add_repeat_channel_node() local
140 const std::vector<int64_t>& in_sizes = t_in->sizes(); in add_repeat_node() local
|
| H A D | Slice.cpp | 70 const auto in_sizes = t_in->sizes(); in add_slice_tensor_copy_node() local
235 std::vector<int64_t> in_sizes = graph.sizes_of(in_ref); in check_slice_view_args() local
|
| H A D | Convolution.cpp | 71 const std::vector<int64_t>& in_sizes = self->sizes(); in resize_conv1d_node() local
434 std::vector<int64_t> in_sizes = t_in->sizes(); in add_conv1d_node() local
|
| H A D | RotaryEmbedding.cpp | 23 std::vector<int64_t> in_sizes = in->sizes(); in resize_rotary_embedding_node() local
|
| H A D | GridPriors.cpp | 28 std::vector<int64_t> in_sizes = in->sizes(); in resize_grid_priors_node() local
|
| H A D | Softmax.cpp | 28 std::vector<int64_t> in_sizes = in->sizes(); in resize_softmax_node() local
|
| H A D | BatchNorm.cpp | 49 std::vector<int64_t> in_sizes = graph.get_tensor(in_ref)->sizes(); in add_native_batch_norm_node() local
|
| H A D | Copy.cpp | 77 std::vector<int64_t> in_sizes = t_in->sizes(); in add_copy_channel_offset_node() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | DepthwiseConv2d.cu | 374 auto in_sizes = input.sizes(); in conv_depthwise2d_forward_out() local
474 auto in_sizes = input.sizes(); in conv_depthwise2d_backward_out() local
601 auto in_sizes = input.sizes(); in conv_depthwise2d_grad_weight_out() local
|
| H A D | MultiMarginLoss.cu | 225 auto in_sizes = input.sizes(); in multi_margin_loss_cuda_out() local
363 auto in_sizes = input.sizes(); in multi_margin_loss_cuda_backward_out() local
|
| H A D | ConvolutionMM2d.cu | 44 const auto in_sizes = input.sizes(); in slow_conv2d_shape_check() local
141 auto in_sizes = input.sizes(); in slow_conv2d_forward() local
|
| H A D | SpectralOps.cpp | 420 auto in_sizes = self.sizes(); in _fft_c2r_cufft() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_convolution.cpp | 39 SizesArrayRef in_sizes, in conv2d_impl()
203 SizesArrayRef in_sizes = in.sizes(); in convolution_wrapper() local
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/glsl/ |
| H A D | broadcasting_utils.h | 9 ivec4 broadcast_indices(const ivec4 out_idx, const ivec4 in_sizes) { in broadcast_indices()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | ragged_range_op.cc | 55 std::vector<int> in_sizes; in Compute() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkl/ |
| H A D | SpectralOps.cpp | 262 auto in_sizes = self.sizes(); in _fft_c2r_mkl() local
532 auto in_sizes = input.sizes(); in _fft_c2r_mkl() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | MaskedFill.cpp | 29 auto in_sizes = self_arg.sizes(); in masked_fill_scalar() local
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| H A D | ragged_range_tflite.cc | 73 std::vector<int> in_sizes; in EvalT() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | kernel_ops_util.h | 145 const exec_aten::ArrayRef<exec_aten::SizesType> in_sizes, in kernel_reduction_then_map_2d()
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 4118 start_index, end_index, kernel_maxes, in_sizes, out_sizes, pooling_fn argument
4159 start_index, end_index, kernel_maxes, in_sizes, out_sizes, pooling_fn argument
|
| /aosp_15_r20/external/bcc/libbpf-tools/powerpc/ |
| H A D | vmlinux.h | 46163 u32 in_sizes[5]; member
|