| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | multithreaded_conv.h | 88 int input_batches, int input_height, int input_width, in operator()
102 EigenMatrix output(output_data, input_batches * conv_width, filter_count); in operator()
103 ConstEigenMatrix input(input_data, input_batches * conv_width, in operator()
116 EigenMatrix output(output_data, input_batches, filter_count); in operator()
117 ConstEigenMatrix input(input_data, input_batches, k); in operator()
122 EigenTensor output(output_data, input_batches, output_height, in operator()
124 ConstEigenTensor input(input_data, input_batches, input_height, in operator()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_benchmark_cpu_test.cc | 32 int input_batches, int input_height, int input_width, in SpatialConvolution() argument
42 typename Benchmark::Dimensions input_dims(input_batches, input_height, in SpatialConvolution()
59 int input_batches, int input_height, in SpatialConvolutionBackwardInput() argument
70 typename Benchmark::Dimensions input_dims(input_batches, input_height, in SpatialConvolutionBackwardInput()
86 int input_batches, int input_height, in SpatialConvolutionBackwardKernel() argument
97 typename Benchmark::Dimensions input_dims(input_batches, input_height, in SpatialConvolutionBackwardKernel()
106 num_computed_elements * (input_batches * input_height * input_width); in SpatialConvolutionBackwardKernel()
249 int input_batches, int input_height, int input_width, in CuboidConvolution() argument
261 input_batches, input_height, input_width, input_planes, input_depth); in CuboidConvolution()
277 int input_batches, int input_height, in CuboidConvolutionBackwardInput() argument
[all …]
|
| D | conv_ops_using_gemm.cc | 91 int input_batches, int input_height, int input_width, in operator ()() argument
128 for (int batch = 0; batch < input_batches; ++batch) { in operator ()()
214 int input_batches, int input_height, int input_width, in operator ()() argument
219 if ((input_batches <= 0) || (input_width <= 0) || (input_height <= 0) || in operator ()()
222 << input_batches << ", " << input_height << ", " in operator ()()
243 const int m = input_batches * input_height * input_width; in operator ()()
256 const int m = input_batches; in operator ()()
327 const int64 patch_count = (input_batches * output_height * output_width); in operator ()()
|
| D | quantized_conv_ops.cc | 55 int input_batches, int input_height, int input_width, in operator ()() argument
102 for (int batch = 0; batch < input_batches; ++batch) { in operator ()()
201 int input_batches, int input_height, int input_width, in operator ()() argument
222 conv_functor(context, input_data, input_batches, input_height, in operator ()()
292 const int64 patch_count = (input_batches * output_height * output_width); in operator ()()
|
| D | conv_ops_fused_image_transform.cc | 261 int input_batches, int resized_height, int resized_width, in operator ()() argument
269 if ((input_batches <= 0) || (padded_width <= 0) || (padded_height <= 0) || in operator ()()
272 << input_batches << ", " << padded_height << ", " in operator ()()
427 for (int batch = 0; batch < input_batches; ++batch) { in operator ()()
575 ((batch == (input_batches - 1)) && in operator ()()
|
| D | eigen_spatial_convolutions_test.cc | 1381 int input_batches, int input_cols, int input_rows, in PackRhsHelper() argument
1400 Dimensions input_dims(input_depth, input_rows, input_cols, input_batches); in PackRhsHelper()
1513 reshape_dims[1] = output_rows * output_cols * input_batches; // num_patches in PackRhsHelper()
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_image_patch.cpp | 188 int input_batches = 1; in test_patch_padding_valid() local
191 Tensor<float, 4> tensor(input_depth, input_rows, input_cols, input_batches); in test_patch_padding_valid()
203 VERIFY_IS_EQUAL(result.dimension(4), input_batches); // number of batches in test_patch_padding_valid()
229 for (int b = 0; b < input_batches; ++b) { // batch in test_patch_padding_valid()
264 int input_batches = 2; in test_patch_padding_valid_same_value() local
268 Tensor<float, 4> tensor(input_depth, input_rows, input_cols, input_batches); in test_patch_padding_valid_same_value()
276 VERIFY_IS_EQUAL(result.dimension(4), input_batches); // number of batches in test_patch_padding_valid_same_value()
302 for (int b = 0; b < input_batches; ++b) { // batch in test_patch_padding_valid_same_value()
337 int input_batches = 1; in test_patch_padding_same() local
341 Tensor<float, 4> tensor(input_depth, input_rows, input_cols, input_batches); in test_patch_padding_same()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops_test.cc | 52 : input_batches(n), in Conv2DDimensions()
60 int input_batches; member
77 return GetRandomTensor({dims.input_batches, dims.input_height, in GetRandomInputTensor()
89 return GetRandomTensor({dims.input_batches, dims.input_height, in GetRandomOutputTensor()
95 return test::AsTensor<int32>({dims.input_batches, dims.input_height, in GetInputSizesTensor()
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 295 input_batches = [dims[0] for dims in dims_list]
296 assert max(input_batches) == min(input_batches)
297 batch_list.append(input_batches[0])
|