Searched refs:dims_data (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 145 RuntimeShape(int dimensions_count, const int32* dims_data) : size_(0) { in RuntimeShape() argument
146 ReplaceWith(dimensions_count, dims_data); in RuntimeShape()
208 inline void ReplaceWith(int dimensions_count, const int32* dims_data) { in ReplaceWith() argument
211 std::memcpy(dst_dims, dims_data, dimensions_count * sizeof(int32)); in ReplaceWith()
244 const int* dims_data = DimsData(); in FlatSize() local
246 const int dim = dims_data[i]; in FlatSize()
359 const int* dims_data = shape.DimsDataUpTo4D(); in Offset() local
360 TFLITE_DCHECK(i0 >= 0 && i0 < dims_data[0]); in Offset()
361 TFLITE_DCHECK(i1 >= 0 && i1 < dims_data[1]); in Offset()
362 TFLITE_DCHECK(i2 >= 0 && i2 < dims_data[2]); in Offset()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 148 RuntimeShape(int dimensions_count, const int32* dims_data) : size_(0) { in RuntimeShape() argument
149 ReplaceWith(dimensions_count, dims_data); in RuntimeShape()
223 inline void ReplaceWith(int dimensions_count, const int32* dims_data) { in ReplaceWith() argument
226 std::memcpy(dst_dims, dims_data, dimensions_count * sizeof(int32)); in ReplaceWith()
259 const int* dims_data = DimsData(); in FlatSize() local
261 const int dim = dims_data[i]; in FlatSize()
375 const int* dims_data = shape.DimsDataUpTo4D(); in Offset() local
376 TFLITE_DCHECK(i0 >= 0 && i0 < dims_data[0]); in Offset()
377 TFLITE_DCHECK(i1 >= 0 && i1 < dims_data[1]); in Offset()
378 TFLITE_DCHECK(i2 >= 0 && i2 < dims_data[2]); in Offset()
[all …]
|
| D | tensor_ctypes.h | 106 const int32_t* dims_data = dims->data; in GetTensorShape() local
107 return RuntimeShape(dims_size, dims_data); in GetTensorShape()
|