Searched refs:DimsToShape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 61 DepthwiseConv(op_params, DimsToShape(input_dims), input_data, in DepthwiseConv()
62 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
63 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
138 DepthwiseConv(op_params, DimsToShape(input_dims), input_data, in DepthwiseConv()
139 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
140 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
223 Conv(op_params, DimsToShape(input_dims), input_data, DimsToShape(filter_dims), in Conv()
224 filter_data, DimsToShape(bias_dims), bias_data, DimsToShape(output_dims), in Conv()
225 output_data, DimsToShape(im2col_dims), im2col_data); in Conv()
303 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 DepthwiseConv(op_params, DimsToShape(input_dims), input_data, in DepthwiseConv()
161 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
162 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
237 DepthwiseConv(op_params, DimsToShape(input_dims), input_data, in DepthwiseConv()
238 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
239 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
309 DimsToShape(bias_dims), bias_data, in AddBiasAndEvalActivationFunction()
310 DimsToShape(array_dims), array_data); in AddBiasAndEvalActivationFunction()
337 FullyConnected(op_params, DimsToShape(input_dims), input_data, in FullyConnected()
338 DimsToShape(weights_dims), weights_data, in FullyConnected()
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 364 DimsToShape(bias_dims), bias_data, in AddBiasAndEvalActivationFunction()
365 DimsToShape(array_dims), array_data); in AddBiasAndEvalActivationFunction()
941 FullyConnected(op_params, DimsToShape(input_dims), input_data, in FullyConnected()
942 DimsToShape(weights_dims), weights_data, in FullyConnected()
943 DimsToShape(bias_dims), bias_data, DimsToShape(output_dims), in FullyConnected()
1227 FullyConnected(op_params, DimsToShape(input_dims), input_data, in FullyConnected()
1228 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in FullyConnected()
1229 bias_data, DimsToShape(output_dims), output_data, in FullyConnected()
1339 FullyConnected(op_params, DimsToShape(input_dims), input_data, in FullyConnected()
1340 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in FullyConnected()
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 296 inline RuntimeShape DimsToShape(const tflite::Dims<4>& dims) { in DimsToShape() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 312 inline RuntimeShape DimsToShape(const tflite::Dims<4>& dims) { in DimsToShape() function
|