| /aosp_15_r20/frameworks/base/media/mca/filterpacks/native/imageproc/ |
| H A D | to_rgba.c | 22 const int* input_sizes, in gray_to_rgb_process()
53 const int* input_sizes, in rgba_to_rgb_process()
85 const int* input_sizes, in gray_to_rgba_process()
117 const int* input_sizes, in rgb_to_rgba_process()
|
| H A D | invert.c | 22 const int* input_sizes, in invert_process()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | pooling_ops_test.py | 70 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
106 def _VerifyValues(self, pool_func, input_sizes, ksize, strides, padding, argument
294 input_sizes, argument
445 input_sizes, argument
|
| H A D | depthwise_conv_op_test.py | 457 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
493 input_sizes, argument
566 def _CompareBackpropInputWithDilation(self, input_sizes, filter_sizes, argument
626 input_sizes, argument
|
| H A D | conv2d_test.py | 41 input_sizes=None, argument
235 input_sizes=None, argument
549 input_sizes=None, argument
|
| H A D | pooling_ops_3d_test.py | 42 def _VerifyValues(self, pool_func, input_sizes, window, strides, padding, argument
189 input_sizes, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/ |
| H A D | NestedTensorTransformerFunctions.cu | 43 const int* input_sizes, in remove_padding_transform0213_2()
84 const int* input_sizes, in remove_padding_2()
117 const int* input_sizes, in remove_padding()
155 const int* input_sizes, in remove_padding_kernelLauncher()
189 const int* input_sizes, in remove_padding_transform0213_kernelLauncher()
253 const int* input_sizes, in add_padding_1()
291 const int* input_sizes, in add_padding_2()
335 const int* input_sizes, in add_padding_3()
387 const int* input_sizes, in add_padding_kernelLauncher()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | conv_ops_test.py | 912 input_sizes, argument
961 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
1135 input_sizes, argument
1180 def _CompareBackFilter(self, input_sizes, filter_sizes, output_sizes, argument
1316 def _RunAndVerifyBackpropInputDilation(self, input_sizes, filter_sizes, argument
1362 def _RunAndVerifyBackpropFilterDilation(self, input_sizes, filter_sizes, argument
1559 input_sizes, argument
1721 input_sizes, argument
|
| H A D | pooling_ops_3d_test.py | 51 def _VerifyOneTest(self, pool_func, input_sizes, window, strides, padding, argument
92 def _VerifyValues(self, pool_func, input_sizes, window, strides, argument
250 input_sizes, argument
|
| H A D | pooling_ops_test.py | 187 def _VerifyOneType(self, pool_func, input_sizes, ksize, strides, padding, argument
281 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
326 input_sizes, argument
1152 input_sizes, argument
1230 input_sizes, argument
1565 expected_input_backprop, input_sizes, output_sizes, argument
|
| H A D | depthwise_conv_op_base.py | 955 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
1015 def _CompareBackpropFilter(self, input_sizes, filter_sizes, output_sizes, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/glsl/ |
| H A D | texel_access.h | 9 ivec4 input_sizes) { in map_output_pos_to_input_pos()
28 ivec4 input_sizes, in load_texel()
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/quantization/ |
| H A D | quantization.cpp | 49 const auto input_sizes = input.sizes(); in _float_to_bfloat16_cpu() local
68 const auto input_sizes = input.sizes(); in _bfloat16_to_float_cpu() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | CUDASparseDescriptors.cpp | 62 IntArrayRef input_sizes = input.sizes(); in createRawDnMatDescriptor() local
145 IntArrayRef input_sizes = input.sizes(); in CuSparseSpMatCsrDescriptor() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ |
| H A D | ts_native_functions.cpp | 441 c10::SymIntArrayRef input_sizes, in select_backward_symint()
506 at::SymIntArrayRef input_sizes, in diagonal_backward_symint()
516 at::SymIntArrayRef input_sizes, in slice_backward_symint()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | LegacyBatchingRegistrations.cpp | 316 Tensor trace_backward_batching_rule(const Tensor& grad, IntArrayRef input_sizes) { in trace_backward_batching_rule()
368 static int64_t getGradInputPhysicalDim(int64_t dim, IntArrayRef input_sizes, int64_t num_batch_dims… in getGradInputPhysicalDim()
372 Tensor select_backward_batching_rule(const Tensor& grad, IntArrayRef input_sizes, int64_t dim, int6… in select_backward_batching_rule()
392 Tensor slice_backward_batching_rule(const Tensor& grad, IntArrayRef input_sizes, int64_t dim, int64… in slice_backward_batching_rule()
408 Tensor diagonal_backward_batching_rule(const Tensor& grad, IntArrayRef input_sizes, int64_t offset,… in diagonal_backward_batching_rule()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesViews.cpp | 322 c10::SymIntArrayRef input_sizes, int64_t offset, int64_t dim1, int64_t dim2) { in diagonal_backward_batch_rule()
394 c10::SymIntArrayRef input_sizes, int64_t dim, c10::SymInt index) { in select_backward_batch_rule()
407 SymIntArrayRef input_sizes, int64_t dim, c10::SymInt start, c10::SymInt end, c10::SymInt step) { in slice_backward_batch_rule()
|
| H A D | PyTorchOperatorHacks.cpp | 56 const auto input_sizes = input.sizes(); in linear_hack() local
133 auto input_sizes = input.sizes(); in make_feature_noise() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | UpSampleMoreKernel.cpp | 104 auto input_sizes = grad_input.sizes().vec(); in cpu_upsample_nearest_backward() local
242 auto input_sizes = grad_input.sizes().vec(); in cpu_upsample_nearest_backward_channels_last() local
435 auto input_sizes = grad_input.sizes().vec(); in cpu_upsample_linear_backward() local
613 auto input_sizes = grad_input.sizes().vec(); in cpu_upsample_linear_backward_channels_last() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | convolution_group_converter.cc | 241 std::vector<int64_t> input_sizes(activation->shape().dimensions().begin(), in HandleBatchGroupCount() local
588 std::vector<int64_t> input_sizes(activation->shape().dimensions().begin(), in HandleConvolution() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LinearAlgebraUtils.h | 583 IntArrayRef input_sizes = input.sizes(); in is_blas_compatible_column_major_order() local
604 IntArrayRef input_sizes = input.sizes(); in is_blas_compatible_row_major_order() local
|
| H A D | UnfoldBackward.cpp | 19 IntArrayRef input_sizes, in unfold_backward()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_conv_ops_test.cc | 157 Node* input_sizes = in DefaultConv2DBwdInput() local
184 Node* input_sizes = in MklConv2DBwdInput() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ConvolutionMM2d.cu | 236 auto input_sizes = input.sizes(); in slow_conv2d_backward() local
308 auto input_sizes = input.sizes(); in slow_conv2d_grad_weight() local
|
| /aosp_15_r20/external/pytorch/test/distributed/flight_recorder/ |
| H A D | test_fr_analysis.py | 23 input_sizes, argument
|