Home
last modified time | relevance | path

Searched refs:DimsToShape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h66 DepthwiseConv(op_params, DimsToShape(input_dims), input_data, in DepthwiseConv()
67 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
68 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
143 DepthwiseConv(op_params, DimsToShape(input_dims), input_data, in DepthwiseConv()
144 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
145 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
228 Conv(op_params, DimsToShape(input_dims), input_data, DimsToShape(filter_dims), in Conv()
229 filter_data, DimsToShape(bias_dims), bias_data, DimsToShape(output_dims), in Conv()
230 output_data, DimsToShape(im2col_dims), im2col_data); in Conv()
308 Conv(op_params, DimsToShape(input_dims), input_data, DimsToShape(filter_dims), in Conv()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h161 const RuntimeShape output_shape = DimsToShape(output_dims); in DepthwiseConv()
164 DepthwiseConvImpl(op_params, DimsToShape(input_dims), input_data, in DepthwiseConv()
165 DimsToShape(filter_dims), filter_data, in DepthwiseConv()
166 DimsToShape(bias_dims), bias_data, output_shape, in DepthwiseConv()
311 const RuntimeShape output_shape = DimsToShape(output_dims); in DepthwiseConv()
315 op_params, DimsToShape(input_dims), input_data, DimsToShape(filter_dims), in DepthwiseConv()
316 filter_data, DimsToShape(bias_dims), bias_data, DimsToShape(output_dims), in DepthwiseConv()
614 DimsToShape(bias_dims), bias_data, in AddBiasAndEvalActivationFunction()
615 DimsToShape(array_dims), array_data); in AddBiasAndEvalActivationFunction()
696 FullyConnected(op_params, DimsToShape(input_dims), input_data, in FullyConnected()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Druntime_shape.h207 inline RuntimeShape DimsToShape(const tflite::Dims<4>& dims) { in DimsToShape() function