Home
last modified time | relevance | path

Searched defs:input_width (Results 1 – 25 of 55) sorted by relevance

123

/external/tensorflow/tensorflow/lite/kernels/internal/
Dresize_bilinear_test.cc28 void TestOneResizeBilinear(int batch, int depth, int input_width, in TestOneResizeBilinear()
84 const int input_width = ExponentialRandomPositiveInt(0.9f, 20, 200); in TEST() local
100 const int input_width = ExponentialRandomPositiveInt(0.9f, 20, 200); in TEST() local
116 const int input_width = ExponentialRandomPositiveInt(0.9f, 20, 200); in TEST() local
132 const int input_width = ExponentialRandomPositiveInt(0.9f, 20, 200); in TEST() local
Dresize_nearest_neighbor_test.cc152 void TestOptimizedResizeNearestNeighbor(int batch, int depth, int input_width, in TestOptimizedResizeNearestNeighbor()
184 bool is_valid_scale(int input_width, int input_height, int output_width, in is_valid_scale()
221 const int input_width = ExponentialRandomPositiveInt(0.9f, 20, 200); in TEST() local
Dsoftmax_quantized_test.cc155 const int input_width = ExponentialRandomPositiveInt(0.8f, 20, 200); in TryOneUniformSoftmax() local
188 const int input_width = ExponentialRandomPositiveInt(0.7f, 20, 200); in TryOneSkyscraperSoftmax() local
Ddepthwiseconv_quantized_test.cc490 int input_depth, int input_width, int input_height, in TryTestDepthwiseConv()
563 const int input_width = ExponentialRandomPositiveInt(0.9f, 20, 200); in TryTestOneDepthwiseConv() local
585 int input_width = ExponentialRandomPositiveInt(0.9f, 20, 200); in TryTestOneDepthwiseConv3x3Filter() local
632 const int input_width = coverage_extension == CoverageExtension::kLargeWidths in TryTestOneNeonDot3x3() local
Dtest_util.cc28 const int input_width = input_shape.Dims(2); in ComputeConvSizes() local
/external/tensorflow/tensorflow/examples/label_image/
Dlabel_image.py40 input_width=299, argument
81 input_width = 299 variable
108 input_width = args.input_width variable
Dmain.cc116 const int input_width, const float input_mean, in ReadTensorFromImageFile()
286 int32 input_width = 299; in main() local
/external/tensorflow/tensorflow/core/kernels/
Ddepthtospace_op_gpu.cu.cc37 const int input_height, const int input_width, in D2S_NHWC()
67 const int block_size, const int input_width, in D2S_NCHW()
104 const int input_width, const int output_width, in D2S_NCHW_LOOP()
153 const int input_width = input.dimension(2); in operator ()() local
184 const int input_width = input.dimension(3); in operator ()() local
Dquantized_activation_ops_test.cc47 const int input_width = 2; in TEST_F() local
78 const int input_width = 2; in TEST_F() local
Dmkl_requantize_ops_test.cc118 const int input_width = 4; in TEST_F() local
179 const int input_width = 4; in TEST_F() local
237 const int input_width = 4; in TEST_F() local
Dspacetodepth_op_gpu.cu.cc35 const int input_height, const int input_width, in S2D_NHWC()
104 const int output_width, const int input_width, in S2D_NCHW_LOOP()
149 const int input_width = input.dimension(2); in operator ()() local
188 const int input_width = input.dimension(3); in operator ()() local
Dquantized_pooling_ops_test.cc54 const int input_width = 4; in TEST_F() local
99 const int input_width = 4; in TEST_F() local
Dquantized_bias_add_op_test.cc54 const int input_width = 3; in TEST_F() local
104 const int input_width = 64; in TEST_F() local
Dmkl_quantized_pooling_ops_test.cc84 const int input_width = 4; in TEST_F() local
153 const int input_width = 4; in TEST_F() local
Deigen_benchmark_cpu_test.cc32 int input_batches, int input_height, int input_width, in SpatialConvolution()
61 int input_width, int input_depth, in SpatialConvolutionBackwardInput()
89 int input_width, int input_depth, in SpatialConvolutionBackwardKernel()
253 int input_batches, int input_height, int input_width, in CuboidConvolution()
283 int input_width, int input_planes, in CuboidConvolutionBackwardInput()
312 int input_width, int input_planes, in CuboidConvolutionBackwardKernel()
Dquantized_batch_norm_op_test.cc65 const int input_width = 6; in TEST_F() local
162 const int input_width = 6; in TEST_F() local
Dscale_and_translate_op.cc305 const int64 input_width = input.dim_size(2); in Compute() local
485 const int64 input_width, const int64 output_height, in GatherColumns()
528 const int64 input_width, const int64 output_height, in GatherRows()
567 const int64 input_width = images.dimension(2); in operator ()() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dpooling.h34 const int input_width = input_shape.Dims(2); in AveragePool() local
95 const int input_width = input_shape.Dims(2); in MaxPool() local
Dmean.h41 const int input_width = input_shape.Dims(2); in Mean() local
Dconv.h59 const int input_width = input_shape.Dims(2); in ConvPerChannel() local
Ddepthwise_conv.h53 const int input_width = input_shape.Dims(2); in DepthwiseConvPerChannel() local
/external/tensorflow/tensorflow/lite/kernels/
Dspace_to_depth.cc59 const int input_width = input->dims->data[2]; in Prepare() local
/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model_test.cc33 const int input_width = 400; in CreateTestGraph() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddepthwiseconv_float.h47 const int input_width = input_shape.Dims(2); in DepthwiseConv() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dmultithreaded_conv.h157 const int input_width = input_shape.Dims(2); in Conv() local

123