| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | python_op_gen.cc | 103 void Unflatten(const string& prefix, const std::vector<string>& output_sizes, in Unflatten()
420 std::vector<string> output_sizes(num_outs_); in Code() local
529 const string& function_setup, const std::vector<string>& output_sizes) { in HandleGraphMode()
778 std::vector<string>* output_sizes, string* num_outputs_expr) { in GetOutputSizesAndNumOutputsExpr()
818 const string& indentation, const std::vector<string>& output_sizes, in AddEagerFunctionTeardown()
853 const string& parameters, const std::vector<string>& output_sizes, in AddEagerFastPathAndGraphCode()
903 const string& parameters, const std::vector<string>& output_sizes, in AddEagerFallbackCode()
1014 std::vector<string> output_sizes; in AddEagerInferredAttrs() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | ConvPrepack.cpp | 105 IntArrayRef output_sizes, in _mkldnn_convolution_out()
154 IntArrayRef output_sizes, in mkldnn_convolution_out()
196 std::vector<int64_t> output_sizes = get_output_sizes(context, input); in run() local
238 std::vector<int64_t> output_sizes = get_output_sizes(context, input); in run() local
|
| H A D | Conv.cpp | 169 std::vector<int64_t>& output_sizes, in _mkldnn_convolution_out()
249 …auto output_sizes = conv_output_size(input_t.sizes(), weight_t.sizes(), padding_expanded, stride_e… in _mkldnn_convolution() local
371 auto output_sizes = conv_output_size( in mkldnn_convolution_pointwise_binary() local
541 auto output_sizes = conv_output_size( in mkldnn_convolution_pointwise_binary_() local
658 …auto output_sizes = conv_input_size(input.sizes(), weight_IOHW_sizes, padding_expanded, output_pad… in _mkldnn_convolution_transpose() local
764 …auto output_sizes = conv_input_size(input_t.sizes(), weight_IOHW_sizes, padding_expanded , output_… in mkldnn_convolution_transpose_pointwise_meta() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | depthwise_conv_op_test.py | 457 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
495 output_sizes, argument
567 output_sizes, stride, dilation, argument
628 output_sizes, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | conv_ops_test.py | 914 output_sizes, argument
961 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
1137 output_sizes, argument
1180 def _CompareBackFilter(self, input_sizes, filter_sizes, output_sizes, argument
1317 output_sizes, strides, dilations, argument
1363 output_sizes, strides, dilations, argument
1561 output_sizes, argument
1723 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
|
| H A D | pooling_ops_test.py | 1153 output_sizes, argument
1231 output_sizes, argument
1565 expected_input_backprop, input_sizes, output_sizes, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/glsl/ |
| H A D | texel_access.h | 8 ivec4 output_sizes, in map_output_pos_to_input_pos()
27 ivec4 output_sizes, in load_texel()
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Upsample.cpp | 27 const ValueRef output_sizes = extra_args[0]; // HW in resize_upsample_nearest2d_node() local
54 const ValueRef output_sizes, in add_upsample_nearest2d_node()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | PixelShuffle.cpp | 27 std::vector<int64_t> output_sizes(self.sizes().begin(), self.sizes().end() - 3); in pixel_shuffle_cpu() local
55 std::vector<int64_t> output_sizes(self.sizes().begin(), self.sizes().end() - 3); in pixel_unshuffle_cpu() local
|
| H A D | FunctionOfAMatrixUtils.cpp | 32 auto output_sizes = input.sizes().vec(); in _compute_linear_combination() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Upsample.cpp | 14 const IntArrayRef output_sizes, in upsample_nearest2d()
99 const IntArrayRef output_sizes, in upsample_bilinear2d()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/ |
| H A D | NestedTensorTransformerFunctions.cu | 44 const int* output_sizes, in remove_padding_transform0213_2()
85 const int* output_sizes, in remove_padding_2()
118 const int* output_sizes, in remove_padding()
156 const int* output_sizes, in remove_padding_kernelLauncher()
190 const int* output_sizes, in remove_padding_transform0213_kernelLauncher()
389 const std::vector<int64_t>& output_sizes, in add_padding_kernelLauncher()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | UpSampleMoreKernel.cpp | 105 auto output_sizes = grad_output.sizes().vec(); in cpu_upsample_nearest_backward() local
243 auto output_sizes = grad_output.sizes().vec(); in cpu_upsample_nearest_backward_channels_last() local
436 auto output_sizes = grad_output.sizes().vec(); in cpu_upsample_linear_backward() local
614 auto output_sizes = grad_output.sizes().vec(); in cpu_upsample_linear_backward_channels_last() local
|
| /aosp_15_r20/external/pytorch/test/distributed/flight_recorder/ |
| H A D | test_fr_analysis.py | 24 output_sizes, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | Functional.cpp | 263 std::vector<int64_t> output_sizes = input.sizes().vec(); in all_to_all_single() local
610 std::vector<int64_t> output_sizes = input.sizes().vec(); in shard_dim_alltoall() local
|
| /aosp_15_r20/external/executorch/kernels/optimized/cpu/ |
| H A D | op_mm.cpp | 29 std::array<exec_aten::SizesType, kTensorDimensionLimit> output_sizes; in opt_mm_out() local
|
| H A D | op_linear.cpp | 36 std::array<exec_aten::SizesType, kTensorDimensionLimit> output_sizes; in opt_linear_out() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_mm.cpp | 27 exec_aten::SizesType output_sizes[kTensorDimensionLimit]; in mm_out() local
|
| H A D | op_bmm.cpp | 32 exec_aten::SizesType output_sizes[kTensorDimensionLimit]; in bmm_out() local
|
| H A D | op_addmm.cpp | 37 exec_aten::SizesType output_sizes[kTensorDimensionLimit]; in addmm_out() local
|
| H A D | op_expand_copy.cpp | 73 exec_aten::SizesType output_sizes[kTensorDimensionLimit]; in expand_copy_out() local
|
| H A D | op_max_pool2d_with_indices.cpp | 43 exec_aten::SizesType output_sizes[kTensorDimensionLimit]; in max_pool2d_with_indices_out() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | split_benchmark.py | 30 def build_graph(device, input_shape, output_sizes, axis): argument
|
| /aosp_15_r20/external/executorch/kernels/quantized/cpu/ |
| H A D | op_mixed_mm.cpp | 64 exec_aten::SizesType output_sizes[kTensorDimensionLimit]; in quantized_mixed_mm_out() local
|