Searched refs:DimsToShape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 64 DepthwiseConv(op_params, DimsToShape(input_dims), input_data, in DepthwiseConv()
65 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
66 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
141 DepthwiseConv(op_params, DimsToShape(input_dims), input_data, in DepthwiseConv()
142 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
143 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
226 Conv(op_params, DimsToShape(input_dims), input_data, DimsToShape(filter_dims), in Conv()
227 filter_data, DimsToShape(bias_dims), bias_data, DimsToShape(output_dims), in Conv()
228 output_data, DimsToShape(im2col_dims), im2col_data); in Conv()
306 Conv(op_params, DimsToShape(input_dims), input_data, DimsToShape(filter_dims), in Conv()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 160 const RuntimeShape output_shape = DimsToShape(output_dims); in DepthwiseConv()
163 DepthwiseConvImpl(op_params, DimsToShape(input_dims), input_data, in DepthwiseConv()
164 DimsToShape(filter_dims), filter_data, in DepthwiseConv()
165 DimsToShape(bias_dims), bias_data, output_shape, in DepthwiseConv()
310 const RuntimeShape output_shape = DimsToShape(output_dims); in DepthwiseConv()
314 op_params, DimsToShape(input_dims), input_data, DimsToShape(filter_dims), in DepthwiseConv()
315 filter_data, DimsToShape(bias_dims), bias_data, DimsToShape(output_dims), in DepthwiseConv()
613 DimsToShape(bias_dims), bias_data, in AddBiasAndEvalActivationFunction()
614 DimsToShape(array_dims), array_data); in AddBiasAndEvalActivationFunction()
695 FullyConnected(op_params, DimsToShape(input_dims), input_data, in FullyConnected()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 342 inline RuntimeShape DimsToShape(const tflite::Dims<4>& dims) { in DimsToShape() function
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 505 inline RuntimeShape DimsToShape(const tflite::Dims<4>& dims) {
|