Searched refs:DimsData (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | tensor_test.cc | 27 std::vector<int32>(d.DimsData(), d.DimsData() + d.DimensionsCount()), in TEST()
34 std::vector<int32>(d.DimsData(), d.DimsData() + d.DimensionsCount()), in TEST()
41 std::vector<int32>(d.DimsData(), d.DimsData() + d.DimensionsCount()), in TEST()
48 std::vector<int32>(d.DimsData(), d.DimsData() + d.DimensionsCount()), in TEST()
|
| D | types.h | 170 std::memcpy(DimsData(), other.DimsData(), sizeof(int32) * size_); in RuntimeShape()
175 std::memcmp(DimsData(), comp.DimsData(), size_ * sizeof(int32)) == 0;
204 inline int32* DimsData() { in DimsData() function
207 inline const int32* DimsData() const { in DimsData() function
233 int32* dst_dims = DimsData(); in ReplaceWith()
242 int32* data = DimsData(); in BuildFrom()
267 const int* dims_data = reinterpret_cast<const int*>(DimsData()); in FlatSize()
290 std::memcpy(DimsData() + size_increase, shape.DimsData(), in RuntimeShape()
638 const auto* dims_data = shape.DimsData(); in FlatSizeSkipDim()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | concatenation.cc | 94 all_shapes[i].ReplaceWith(shape.DimensionsCount(), shape.DimsData()); in GetAllTensorShapes()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | rfft2d.cc | 333 const auto* input_dims_data = input_shape.DimsData(); in Rfft2dHelper()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 7394 const int d0 = input_shape.DimsData()[0]; in Transpose2D()
7395 const int d1 = input_shape.DimsData()[1]; in Transpose2D()
7492 const int d0 = input_shape.DimsData()[0]; in Transpose2D()
7493 const int d1 = input_shape.DimsData()[1]; in Transpose2D()
|