Home
last modified time | relevance | path

Searched refs:ComputeConvSizes (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dtest_util.cc23 bool ComputeConvSizes(const RuntimeShape& input_shape, int output_depth, in ComputeConvSizes() function
Dtest_util.h30 bool ComputeConvSizes(const RuntimeShape& input_shape, int output_depth,
Ddepthwiseconv_float_test.cc106 if (!ComputeConvSizes(input_shape_inference, output_depth, filter_width, in TryTestOneDepthwiseConv()
Ddepthwiseconv_quantized_test.cc524 if (!ComputeConvSizes(input_shape_inference, output_depth, filter_width, in TryTestDepthwiseConv()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc34 void ComputeConvSizes(const Shape& input_shape, int output_depth, int kwidth, in ComputeConvSizes() function
147 ComputeConvSizes(input_shape, output_depth, kwidth, kheight, op->stride_width, in ProcessConvOperator()
288 ComputeConvSizes(input_shape, output_depth, kwidth, kheight, op->stride_width, in ProcessDepthwiseConvOperator()
892 ComputeConvSizes(input_shape, output_depth, op->kwidth, op->kheight, in ProcessAveragePoolOperator()
909 ComputeConvSizes(input_shape, output_depth, op->kwidth, op->kheight, in ProcessMaxPoolOperator()
928 ComputeConvSizes(input_shape, output_depth, op->kwidth, op->kheight, in ProcessL2PoolOperator()