Home
last modified time | relevance | path

Searched defs:input_depth (Results 1 – 25 of 87) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_backward_spatial_convolutions_test.cc31 const int input_depth = 2; in TEST() local
80 const int input_depth = 2; in TEST() local
132 const int input_depth = 2; in TEST() local
180 const int input_depth = 2; in TEST() local
232 const int input_depth = 2; in TEST() local
287 const int input_depth = 2; in TEST() local
341 const int num_batches, const int input_depth, const int input_rows, in test_batched_strided_spatial_convolution_backward_input_valid()
399 int input_depth = 1; in TEST() local
417 const int num_batches, const int input_depth, const int input_rows, in test_batched_strided_spatial_convolution_backward_input_valid_row_major()
476 int input_depth = 1; in TEST() local
[all …]
H A Deigen_backward_cuboid_convolutions_test.cc31 const int input_depth = 2; in TEST() local
94 const int input_depth = 2; in TEST() local
157 const int input_depth = 2; in TEST() local
228 const int input_depth = 2; in TEST() local
300 const int input_depth = 2; in TEST() local
367 const int input_depth = 2; in TEST() local
433 const int input_depth = 2; in TEST() local
499 const int input_depth = 2; in TEST() local
566 const int input_depth = 2; in TEST() local
633 const int input_depth = 2; in TEST() local
[all …]
H A Deigen_benchmark_cpu_test.cc33 int input_depth, in SpatialConvolution()
60 int input_width, int input_depth, in SpatialConvolutionBackwardInput()
87 int input_width, int input_depth, in SpatialConvolutionBackwardKernel()
250 int input_planes, int input_depth, in CuboidConvolution()
279 int input_depth, in CuboidConvolutionBackwardInput()
307 int input_depth, in CuboidConvolutionBackwardKernel()
H A Deigen_spatial_convolutions_test.cc30 const int input_depth = 7; in TEST() local
75 const int input_depth = 7; in TEST() local
194 const int input_depth = 10; in TEST() local
246 const int input_depth = 10; in TEST() local
304 const int input_depth = 10; in TEST() local
360 const int input_depth = 10; in TEST() local
408 const int input_depth = 2; in TEST() local
456 const int input_depth = 10; in TEST() local
507 const int input_depth = 10; in TEST() local
558 const int input_depth = 10; in TEST() local
[all …]
H A Dspacetodepth_op_gpu.cu.cc36 const int input_depth, const int output_height, in S2D_NHWC()
150 const int input_depth = input.dimension(3); in operator ()() local
178 const int input_depth = input.dimension(1); in operator ()() local
H A Ddepthtospace_op_gpu.cu.cc37 const int input_depth, const int output_height, in D2S_NHWC()
153 const int input_depth = input.dimension(3); in operator ()() local
181 const int input_depth = input.dimension(1); in operator ()() local
H A Dconv_ops_fused_image_transform.cc112 int input_depth; member
171 int64_t cache_line_width, int64_t input_width, int64_t input_depth, in CalculatePerCacheLineParameters()
228 T1* cache_line_start, int64_t input_depth, in CalculatePerCachePixelParameters()
262 int padded_height, int padded_width, int input_depth, in operator ()()
H A Dspacetodepth_op.cc103 const int input_depth = in Compute() local
172 const int input_depth = input.dimension(3); in operator ()() local
H A Dquantized_batch_norm_op_test.cc65 const int input_depth = 2; in TEST_F() local
162 const int input_depth = 2; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Ddepthwise_conv_test.cc108 int input_depth = GetShape(input_)[3]; in BaseDepthwiseConvolutionOpModel() local
478 const int input_depth = 4; in BatchPaddingValidTest() local
539 const int input_depth = 1; in BatchPaddingSameTest() local
1072 const int input_depth = 8; in TEST_P() local
1145 const int input_depth = 8; in TEST_P() local
1220 const int input_depth = 8; in TEST_P() local
1284 const int input_depth = 8; in TEST_P() local
1357 const int input_depth = 8; in TEST_P() local
1421 const int input_depth = 8; in TEST_P() local
1491 const int input_depth = 8; in TEST_P() local
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DMaxPoolKernel.cpp74 int64_t input_depth, int64_t input_height, int64_t input_width, int64_t channels, in compute_internal()
149 int64_t input_depth, int64_t input_height, int64_t input_width, int64_t channels, in compute_internal()
285 int64_t input_depth = is_3d ? input.size(-3) : 1; in cpu_max_pool() local
401 int64_t input_depth = is_3d ? input.size(-3) : 1; in cpu_max_pool_channels_last() local
501 int64_t input_depth = is_3d ? grad_input.size(-3) : 1; in cpu_max_pool_backward() local
562 int64_t input_depth = is_3d ? grad_input.size(2) : 1; in cpu_max_pool_backward_channels_last() local
H A DAdaptiveAvgPoolKernel.cpp426 int64_t input_depth = input.size(-3); in cpu_adaptive_avg_pool3d() local
491 int64_t input_depth = input.size(2); in cpu_adaptive_avg_pool3d_channels_last() local
587 int64_t input_depth = input.size(2); in cpu_adaptive_avg_pool3d_channels_last() local
694 int64_t input_depth = grad_input.size(-3); in cpu_adaptive_avg_pool3d_backward() local
753 int64_t input_depth = grad_input.size(2); in cpu_adaptive_avg_pool3d_backward_channels_last() local
H A DPaddingKernel.cpp146 int64_t input_depth = ndim == 3 ? p.ishape[ndim - 3] : 1; in cpu_padding() local
253 int64_t input_depth = ndim == 3 ? p.ishape[ndim - 3] : 1; in cpu_padding_channels_last() local
327 int64_t input_depth = ndim == 3 ? p.ishape[ndim - 3] : 1; in cpu_padding_backward() local
415 int64_t input_depth = ndim == 3 ? p.ishape[ndim - 3] : 1; in cpu_padding_backward_channels_last() local
H A DAdaptiveMaxPoolKernel.cpp500 int64_t input_depth = input.size(-3); in cpu_adaptive_max_pool3d() local
578 int64_t input_depth = input.size(2); in cpu_adaptive_max_pool3d_channels_last() local
703 int64_t input_depth = input.size(2); in cpu_adaptive_max_pool3d_channels_last() local
847 int64_t input_depth = grad_input.size(-3); in cpu_adaptive_max_pool3d_backward() local
899 int64_t input_depth = grad_input.size(2); in cpu_adaptive_max_pool3d_backward_channels_last() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DUpSampleNearest3d.cpp31 int64_t input_depth, in upsample_nearest3d_out_frame()
88 int64_t input_depth, in upsample_nearest3d_out_frame_nhwc()
155 int64_t input_depth = input.size(2); in _upsample_nearest3d_quantized_cpu() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dim2col_utils.h142 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in DilatedIm2col() local
226 const int input_depth = input_shape.Dims(3); in Im2col() local
263 const int input_depth = input_shape.Dims(3); in Im2col() local
389 const int input_depth = input_shape.Dims(1); in Im2col3D() local
432 const int input_depth = input_shape.Dims(1); in DilatedIm2col3D() local
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_image_patch.cpp185 int input_depth = 3; in test_patch_padding_valid() local
261 int input_depth = 1; in test_patch_padding_valid_same_value() local
334 int input_depth = 3; in test_patch_padding_same() local
411 int input_depth = 1; in test_patch_padding_same_negative_padding_clip_to_zero() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dconv_2d_helpers.cc155 int input_depth, in CheckShouldSplitDwConv()
183 const std::vector<uint8_t>& converted_data, int input_depth, in SplitWeightsForDwConv()
226 TfLiteContext* context, const int input_depth) { in InitializeWeightsNodes()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvolutionMM3d.cpp51 const int64_t input_depth = input.size(dim_depth); in compute_columns3d() local
173 const int64_t input_depth = input.size(dim_depth); in slow_conv3d_shape_check() local
280 int64_t input_depth, in slow_conv3d_update_output_frame()
416 const int64_t input_depth = input.size(dim_depth); in slow_conv3d_backward_out_cpu_template() local
604 const int64_t input_depth = input.size(dim_depth); in slow_conv3d_forward_out_cpu() local
H A DNaiveConvolutionTranspose3d.cpp133 const int64_t input_depth = input.size(dimd); in slow_conv_transpose3d_shape_check() local
267 const int64_t input_depth = input.size(2); in slow_conv_transpose3d_out_cpu_template() local
488 const int64_t input_depth = input.size(2); in slow_conv_transpose3d_backward_out_cpu_template() local
706 const int64_t input_depth = input.size(2); in slow_conv_transpose3d_acc_grad_parameters_cpu() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_conv_ops.h125 int input_depth = static_cast<int>(input_depth_raw); in GetInputSizeInMklOrder() local
208 int input_depth = GetTensorDim(input_shape, data_format_, 'C'); in GetFilterSizeInMklOrder() local
506 int input_depth = GetTensorDim(input_shape, data_format_, 'C'); variable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNaiveConvolutionTranspose3d.cu132 int64_t input_depth = input.size(dimd); in slow_conv_transpose3d_shape_check() local
270 int64_t input_depth = input.size(2); in slow_conv_transpose3d_out_cuda_template() local
501 int64_t input_depth = input.size(2); in slow_conv_transpose3d_backward_out_cuda_template() local
725 int64_t input_depth = input.size(2); in slow_conv_transpose3d_acc_grad_parameters_cuda() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A Ddepthwise_conv.h55 const int input_depth = input_shape.Dims(3); in DepthwiseConvPerChannel() local
152 const int input_depth = input_shape.Dims(3); in DepthwiseConvPerChannel() local
235 const int input_depth = input_shape.Dims(3); in DepthwiseConvHybridPerChannel() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dconv.h48 const int input_depth = input_shape.Dims(3); in Conv() local
138 const int input_depth = input_shape.Dims(3); in Conv() local
225 const int input_depth = input_shape.Dims(3); in HybridConvPerChannel() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Daveragepool_quantized_test.cc66 const int input_depth = UniformRandomInt(1, 700); in CreateDataAndRunAveragePool() local
129 const int input_depth = UniformRandomInt(1, 700); in CreateExtremalDataAndRunAveragePool() local

1234