Home
last modified time | relevance | path

Searched refs:SHAPE_ARRAY_SIZE (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc475 std::array<int64, SHAPE_ARRAY_SIZE> shape_array; in RegisterConstantNode()
488 CHECK_EQ(4, SHAPE_ARRAY_SIZE); in RegisterConstantNode()
542 CHECK_EQ(4, SHAPE_ARRAY_SIZE); in RegisterConstTensor()
543 for (int i = 0; i < SHAPE_ARRAY_SIZE; ++i) { in RegisterConstTensor()
544 if (i < SHAPE_ARRAY_SIZE - dims) { in RegisterConstTensor()
548 tensor.shape().dim_size(i - (SHAPE_ARRAY_SIZE - dims))); in RegisterConstTensor()
618 std::array<int64, SHAPE_ARRAY_SIZE> shape_array; in IsNodeFlattenReshape()
962 /* static */ std::array<int64, GraphTransferer::SHAPE_ARRAY_SIZE>
968 return std::array<int64, SHAPE_ARRAY_SIZE>{{1, 1, 1, 1}}; in BuildShapeArray()
970 return std::array<int64, SHAPE_ARRAY_SIZE>{ in BuildShapeArray()
[all …]
Dgraph_transferer.h49 static constexpr int SHAPE_ARRAY_SIZE = MAX_SUPPORTED_RANK; variable
95 static std::array<int64, SHAPE_ARRAY_SIZE> ToTensorShapeArray(
190 static std::array<int64, SHAPE_ARRAY_SIZE> BuildShapeArray(
Dhexagon_control_wrapper.h64 const std::array<int64, GraphTransferer::SHAPE_ARRAY_SIZE>& shape,
Dhexagon_control_wrapper.cc323 const std::array<int64, GraphTransferer::SHAPE_ARRAY_SIZE>& shape, in FillInputNode()
430 const std::array<int64, GraphTransferer::SHAPE_ARRAY_SIZE> shape = in FillInputNode()