Searched refs:dims_data (Results 1 – 2 of 2) sorted by relevance
156 RuntimeShape(int dimensions_count, const int32* dims_data) : size_(0) { in RuntimeShape() argument157 ReplaceWith(dimensions_count, dims_data); in RuntimeShape()231 inline void ReplaceWith(int dimensions_count, const int32* dims_data) { in ReplaceWith() argument234 std::memcpy(dst_dims, dims_data, dimensions_count * sizeof(int32)); in ReplaceWith()267 const int* dims_data = reinterpret_cast<const int*>(DimsData()); in FlatSize() local269 buffer_size *= dims_data[i]; in FlatSize()381 const int* dims_data = reinterpret_cast<const int*>(shape.DimsDataUpTo4D()); in Offset() local382 TFLITE_DCHECK(i0 >= 0 && i0 < dims_data[0]); in Offset()383 TFLITE_DCHECK(i1 >= 0 && i1 < dims_data[1]); in Offset()384 TFLITE_DCHECK(i2 >= 0 && i2 < dims_data[2]); in Offset()[all …]
41 const int32_t* dims_data = reinterpret_cast<const int32_t*>(dims->data); in GetTensorShape() local42 return RuntimeShape(dims_size, dims_data); in GetTensorShape()