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.h63 DepthwiseConv(op_params, DimsToShape(input_dims), input_data, in DepthwiseConv()
64 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
65 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
140 DepthwiseConv(op_params, DimsToShape(input_dims), input_data, in DepthwiseConv()
141 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
142 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
225 Conv(op_params, DimsToShape(input_dims), input_data, DimsToShape(filter_dims), in Conv()
226 filter_data, DimsToShape(bias_dims), bias_data, DimsToShape(output_dims), in Conv()
227 output_data, DimsToShape(im2col_dims), im2col_data); in Conv()
305 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()
612 DimsToShape(bias_dims), bias_data, in AddBiasAndEvalActivationFunction()
613 DimsToShape(array_dims), array_data); in AddBiasAndEvalActivationFunction()
694 FullyConnected(op_params, DimsToShape(input_dims), input_data, in FullyConnected()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h318 inline RuntimeShape DimsToShape(const tflite::Dims<4>& dims) { in DimsToShape() function