Searched refs:dims_data (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 175 RuntimeShape(int dimensions_count, const int32_t* dims_data) : size_(0) { in RuntimeShape() argument
176 ReplaceWith(dimensions_count, dims_data); in RuntimeShape()
255 inline void ReplaceWith(int dimensions_count, const int32_t* dims_data) { in ReplaceWith() argument
258 std::memcpy(dst_dims, dims_data, dimensions_count * sizeof(int32_t)); in ReplaceWith()
291 const int* dims_data = reinterpret_cast<const int*>(DimsData()); in FlatSize() local
293 buffer_size *= dims_data[i]; in FlatSize()
405 const int* dims_data = reinterpret_cast<const int*>(shape.DimsDataUpTo5D()); in Offset() local
406 TFLITE_DCHECK(i0 >= 0 && i0 < dims_data[0]); in Offset()
407 TFLITE_DCHECK(i1 >= 0 && i1 < dims_data[1]); in Offset()
408 TFLITE_DCHECK(i2 >= 0 && i2 < dims_data[2]); in Offset()
[all …]
|
| D | tensor_ctypes.h | 41 const int32_t* dims_data = reinterpret_cast<const int32_t*>(dims->data); in GetTensorShape() local
42 return RuntimeShape(dims_size, dims_data); in GetTensorShape()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fill_test.cc | 43 std::initializer_list<dims_type> dims_data, in FillOpModel() argument
49 dims_ = AddConstInput(dims_tensor_type, dims_data, dims_shape); in FillOpModel()
58 if (dims_data.size() > 0) { in FillOpModel()
59 PopulateTensor<dims_type>(dims_, dims_data); in FillOpModel()
81 std::initializer_list<dims_type> dims_data, in QuantizedFillOpModel() argument
90 if (dims_data.size() > 0) { in QuantizedFillOpModel()
91 PopulateTensor<dims_type>(dims_, dims_data); in QuantizedFillOpModel()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | dequantize.h | 59 const int32_t* dims_data = input_shape.DimsData(); in PerChannelDequantize() local
64 ReducedOutputOffset(num_dims, reinterpret_cast<const int*>(dims_data), in PerChannelDequantize()
71 } while (NextIndex(num_dims, reinterpret_cast<const int*>(dims_data), in PerChannelDequantize()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 343 RuntimeShape(int dimensions_count, const int32_t* dims_data) : size_(0) {
344 ReplaceWith(dimensions_count, dims_data);
418 inline void ReplaceWith(int dimensions_count, const int32_t* dims_data) {
421 std::memcpy(dst_dims, dims_data, dimensions_count * sizeof(int32_t));
454 const int* dims_data = reinterpret_cast<const int*>(DimsData());
456 buffer_size *= dims_data[i];
568 const int* dims_data = reinterpret_cast<const int*>(shape.DimsDataUpTo5D());
569 TFLITE_DCHECK(i0 >= 0 && i0 < dims_data[0]);
570 TFLITE_DCHECK(i1 >= 0 && i1 < dims_data[1]);
571 TFLITE_DCHECK(i2 >= 0 && i2 < dims_data[2]);
[all …]
|
| D | ceva_tflm_lib.h | 255 const int* dims_data);
549 const int* dims_data, const int* dims_data_inp);
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | kernel_util.cc | 36 const int32_t* dims_data = reinterpret_cast<const int32_t*>(dims->data); in GetTensorShape() local
37 return RuntimeShape(dims_size, dims_data); in GetTensorShape()
|