Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/core/kernels/
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 …]
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 …]
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
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
Dspacetodepth_op.cc103 const int input_depth = in Compute() local
172 const int input_depth = input.dimension(3); in operator ()() local
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()
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
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 …]
/external/tensorflow/tensorflow/lite/kernels/
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 …]
/external/pytorch/aten/src/ATen/native/quantized/cpu/
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
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
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()
/external/tensorflow/tensorflow/core/kernels/mkl/
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
/external/eigen/unsupported/test/
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
/external/tensorflow/tensorflow/lite/kernels/internal/
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
Dsoftmax_quantized_test.cc178 const int input_depth = ExponentialRandomPositiveInt(0.75f, 175, 500); in TryOneUniformSoftmax() local
209 const int input_depth = small_depth in TryOneSkyscraperSoftmax() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
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
Dconv.h53 const int input_depth = input_shape.Dims(3); in ConvPerChannel() local
164 const int input_depth = input_shape.Dims(3); in ConvPerChannel() local
Dtranspose_conv.h45 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv() local
145 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
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
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dspace_to_depth.h38 const int input_depth = input_shape.Dims(3); in SpaceToDepth() local
Ddepth_to_space.h36 const int input_depth = input_shape.Dims(3); in DepthToSpace() local
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
Dtranspose_conv.h44 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv() local
134 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv() local
/external/pytorch/aten/src/ATen/native/
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
/external/pytorch/aten/src/ATen/native/cpu/
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

1234