Home
last modified time | relevance | path

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

12

/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
195 const int input_depth = 2; in TEST() local
258 const int input_depth = 2; in TEST() local
306 const int input_depth = 2; in TEST() local
357 const int input_depth = 2; in TEST() local
428 const int input_depth = 2; in TEST() local
500 const int input_depth = 2; in TEST() local
555 const int input_depth = 2; in TEST() local
[all …]
Dspacetodepth_op.cc88 const int input_depth = in Compute() local
157 const int input_depth = input.dimension(3); 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
Ddepthtospace_op_gpu.cu.cc38 const int input_depth, const int output_height, in D2S_NHWC()
154 const int input_depth = input.dimension(3); in operator ()() local
182 const int input_depth = input.dimension(1); in operator ()() local
Deigen_benchmark_cpu_test.cc33 int input_depth, in SpatialConvolution()
61 int input_width, int input_depth, in SpatialConvolutionBackwardInput()
89 int input_width, int input_depth, in SpatialConvolutionBackwardKernel()
254 int input_planes, int input_depth, in CuboidConvolution()
284 int input_depth, in CuboidConvolutionBackwardInput()
313 int input_depth, in CuboidConvolutionBackwardKernel()
Dquantized_batch_norm_op_test.cc66 const int input_depth = 2; in TEST_F() local
163 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 …]
Dmkl_conv_ops.h121 int input_depth = static_cast<int>(input_depth_raw); in GetInputSizeInMklOrder() local
205 int input_depth = GetTensorDim(input_shape, data_format_, 'C'); in GetFilterSizeInMklOrder() local
Dconv_ops_fused_image_transform.cc111 int input_depth; member
170 int64 input_width, int64 input_depth, int64 top_padding, int64 pad_offset, in CalculatePerCacheLineParameters()
227 T1* cache_line_start, int64 input_depth, in CalculatePerCachePixelParameters()
261 int padded_height, int padded_width, int input_depth, in operator ()()
Ddepthtospace_op.cc88 const int input_depth = in Compute() local
Dconv_ops_using_gemm.cc90 int input_depth, const T2* filter_data, int filter_height, in operator ()()
213 int input_depth, const T2* filter_data, int filter_height, in operator ()()
Dquantized_conv_ops.cc56 int input_depth, int input_offset, const T2* filter_data, in operator ()()
202 int input_depth, int input_offset, const T2* filter_data, in operator ()()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dsoftmax_quantized_test.cc154 const int input_depth = ExponentialRandomPositiveInt(0.75f, 175, 500); in TryOneUniformSoftmax() local
185 const int input_depth = small_depth in TryOneSkyscraperSoftmax() local
Ddepthwiseconv_quantized_test.cc490 int input_depth, int input_width, int input_height, in TryTestDepthwiseConv()
562 const int input_depth = ExponentialRandomPositiveInt(0.9f, 6, 50); in TryTestOneDepthwiseConv() local
584 const int input_depth = 8 * ExponentialRandomPositiveInt(0.9f, 10, 50); in TryTestOneDepthwiseConv3x3Filter() local
629 const int input_depth = test_param.test_depth_multiplier in TryTestOneNeonDot3x3() local
Dlogsoftmax_quantized_test.cc250 const int input_depth = ExponentialRandomPositiveInt(0.75f, 175, 500); in TryOneUniformLogSoftmax() local
281 const int input_depth = small_depth in TryOneSkyscraperLogSoftmax() local
Ddepthwiseconv_float_test.cc72 const int input_depth = ExponentialRandomPositiveInt(0.9f, 6, 50); in TryTestOneDepthwiseConv() 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/reference/
Ddepthwiseconv_float.h48 const int input_depth = input_shape.Dims(3); in DepthwiseConv() local
Ddepthwiseconv_uint8.h122 const int input_depth = input_shape.Dims(3); in Run() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dconv.h51 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in ConvPerChannel() local
Ddepthwise_conv.h54 const int input_depth = input_shape.Dims(3); in DepthwiseConvPerChannel() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dmultithreaded_conv.h154 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in Conv() local
/external/tensorflow/tensorflow/lite/kernels/
Dfully_connected_test.cc229 void ShuffleAndSetWeights(const std::vector<float>& data, int input_depth, in ShuffleAndSetWeights()
549 TfLiteRegistration* registration, int input_depth, int output_depth, in SimpleTestQuantizedInt16OutputCase()
656 int input_depth = 16 * input_depth_numblocks; in TEST_P() local
Dtranspose_conv.cc121 const int input_depth = SizeOfDimension(input, 3); in ResizeIm2ColTensor() local
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Ddepthwise_conv_test.cc66 int input_depth = input_dims->data[3]; in TestDepthwiseConvFloat() local
146 int input_depth = input_dims->data[3]; in TestDepthwiseConvQuantized() local

12