| /external/tensorflow/tensorflow/lite/delegates/coreml/ |
| D | coreml_delegate_kernel.mm | 29 TfLiteStatus GetDims(int* batch_size, int* height_size, int* width_size, int* depth_size,
34 int* dim[] = {batch_size, height_size, width_size, depth_size};
43 int batch_size, height_size, width_size, depth_size;
44 GetDims(&batch_size, &height_size, &width_size, &depth_size, hwc_dims);
45 RuntimeShape hwc_shape({batch_size, height_size, width_size, depth_size});
46 RuntimeShape chw_shape({batch_size, depth_size, height_size, width_size});
52 int batch_size, height_size, width_size, depth_size;
53 GetDims(&batch_size, &height_size, &width_size, &depth_size, hwc_dims);
54 RuntimeShape hwc_shape({batch_size, height_size, width_size, depth_size});
55 RuntimeShape chw_shape({batch_size, depth_size, height_size, width_size});
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | matmul_builder.cc | 32 void GetDims(int* batch_size, int* height_size, int* width_size, in GetDims() argument
34 int* dim[] = {batch_size, height_size, width_size, depth_size}; in GetDims()
166 int batch_size, height_size, width_size, depth_size; in PopulateSubGraph() local
169 GetDims(&batch_size, &height_size, &width_size, &depth_size, in PopulateSubGraph()
171 weights_shape_ = {batch_size, height_size, depth_size, width_size}; in PopulateSubGraph()
172 RuntimeShape nhwc_shape({batch_size, height_size, width_size, depth_size}); in PopulateSubGraph()
173 RuntimeShape nhcw_shape({batch_size, height_size, depth_size, width_size}); in PopulateSubGraph()
228 int batch_size, height_size, width_size, depth_size; in PopulateSubGraph() local
229 GetDims(&batch_size, &height_size, &width_size, &depth_size, in PopulateSubGraph()
231 weights_shape_ = {batch_size, height_size, depth_size, width_size}; in PopulateSubGraph()
|
| D | cast_builder.cc | 43 int batch_size, height_size, width_size, depth_size; in PopulateSubGraph() local
44 GetDims(&batch_size, &height_size, &width_size, &depth_size, tensor.dims); in PopulateSubGraph()
62 {batch_size, height_size, width_size, depth_size}); in PopulateSubGraph()
|
| D | rsqrt_builder.cc | 86 int batch_size, height_size, width_size, depth_size; in PopulateSubGraph() local
87 GetDims(&batch_size, &height_size, &width_size, &depth_size, tensor.dims); in PopulateSubGraph()
90 num_elements_ = batch_size * height_size * width_size * depth_size; in PopulateSubGraph()
102 sizeof(uint8_t), 4, {batch_size, height_size, width_size, depth_size}); in PopulateSubGraph()
|
| D | op_builder.cc | 205 int batch_size, height_size, width_size, depth_size; in AddConstNodeWithData() local
206 GetDims(&batch_size, &height_size, &width_size, &depth_size, tensor.dims); in AddConstNodeWithData()
207 const int shape[] = {batch_size, height_size, width_size, depth_size}; in AddConstNodeWithData()
224 graph_id_, node_id, batch_size, height_size, width_size, depth_size, in AddConstNodeWithData()
327 int batch_size, height_size, width_size, depth_size; in AddOutput() local
328 GetDims(&batch_size, &height_size, &width_size, &depth_size, dims); in AddOutput()
335 max_sizes[1] = height_size; in AddOutput()
|
| D | op_builder.h | 177 void GetDims(int* batch_size, int* height_size, int* width_size, in GetDims() argument
179 int* dim[] = {batch_size, height_size, width_size, depth_size}; in GetDims()
373 void GetDims(int* batch_size, int* height_size, int* width_size, in GetDims() argument
375 int* dim[] = {batch_size, height_size, width_size, depth_size}; in GetDims()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | utils.h | 27 TfLiteStatus Get4DShape(unsigned int* batch_size, unsigned int* height_size,
|
| D | utils.cc | 61 TfLiteStatus Get4DShape(unsigned int* batch_size, unsigned int* height_size, in Get4DShape() argument
65 unsigned int* dim[] = {batch_size, height_size, width_size, depth_size}; in Get4DShape()
|