Searched refs:DimsData (Results 1 – 8 of 8) 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 | 193 std::memcpy(DimsData(), other.DimsData(), sizeof(int32_t) * size_); in RuntimeShape()
198 std::memcmp(DimsData(), comp.DimsData(), size_ * sizeof(int32_t)) ==
228 inline int32_t* DimsData() { in DimsData() function
231 inline const int32_t* DimsData() const { in DimsData() function
257 int32_t* dst_dims = DimsData(); in ReplaceWith()
266 int32_t* data = DimsData(); in BuildFrom()
291 const int* dims_data = reinterpret_cast<const int*>(DimsData()); in FlatSize()
314 std::memcpy(DimsData() + size_increase, shape.DimsData(), in RuntimeShape()
676 const auto* dims_data = shape.DimsData(); in FlatSizeSkipDim()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 357 std::memcpy(DimsData(), other.DimsData(), sizeof(int32_t) * size_);
362 std::memcmp(DimsData(), comp.DimsData(), size_ * sizeof(int32_t)) == 0;
391 inline int32_t* DimsData() {
394 inline const int32_t* DimsData() const {
420 int32_t* dst_dims = DimsData();
429 int32_t* data = DimsData();
454 const int* dims_data = reinterpret_cast<const int*>(DimsData());
477 std::memcpy(DimsData() + size_increase, shape.DimsData(),
825 const auto* dims_data = shape.DimsData();
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | dequantize.h | 59 const int32_t* dims_data = input_shape.DimsData(); in PerChannelDequantize()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | concatenation.cc | 61 all_shapes[i].ReplaceWith(shape.DimensionsCount(), shape.DimsData()); in GetAllInputTensorShapes()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | rfft2d.cc | 370 const auto* input_dims_data = input_shape.DimsData(); in Rfft2dHelper()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| D | ragged_tensor_to_tensor_tflite.cc | 463 output_shape.DimsData() + ragged_rank + 1); in SetOutputT()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 7602 const int d0 = input_shape.DimsData()[0]; in Transpose2D()
7603 const int d1 = input_shape.DimsData()[1]; in Transpose2D()
7700 const int d0 = input_shape.DimsData()[0]; in Transpose2D()
7701 const int d1 = input_shape.DimsData()[1]; in Transpose2D()
|