Home
last modified time | relevance | path

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

123

/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 …]
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_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
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 …]
Dconv_ops_fused_image_transform.cc112 int input_depth; member
171 int64 input_width, int64 input_depth, int64 top_padding, int64 pad_offset, in CalculatePerCacheLineParameters()
228 T1* cache_line_start, int64 input_depth, in CalculatePerCachePixelParameters()
262 int padded_height, int padded_width, int input_depth, in operator ()()
/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/tensorflow/tensorflow/lite/kernels/internal/
Daveragepool_quantized_test.cc64 const int input_depth = UniformRandomInt(1, 700); in CreateDataAndRunAveragePool() local
127 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
Dmaxpool_quantized_test.cc60 const int input_depth = UniformRandomInt(1, 700); in CreateDataAndRunMaxPool() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dconv.h46 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in Conv() local
129 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in Conv() local
209 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in HybridConvPerChannel() local
Dspace_to_depth.h36 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
Dtranspose_conv.h42 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv() local
132 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv() local
Ddepthwiseconv_uint8.h121 const int input_depth = input_shape.Dims(3); in Run() local
208 const int input_depth = input_shape.Dims(3); in RunPerChannel() local
Ddepthwiseconv_float.h48 const int input_depth = input_shape.Dims(3); in DepthwiseConv() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dim2col_utils.h141 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in DilatedIm2col() local
225 const int input_depth = input_shape.Dims(3); in Im2col() local
262 const int input_depth = input_shape.Dims(3); in Im2col() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Ddepthwise_conv.h53 const int input_depth = input_shape.Dims(3); in DepthwiseConvPerChannel() local
150 const int input_depth = input_shape.Dims(3); in DepthwiseConvPerChannel() local
233 const int input_depth = input_shape.Dims(3); in DepthwiseConvHybridPerChannel() local
Dconv.h51 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in ConvPerChannel() local
154 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in ConvPerChannel() local
Dtranspose_conv.h43 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv() local
142 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv() local
/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
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dtranspose_conv.h43 const int input_depth = in TransposeConvV2() local

123