Searched defs:shape_data (Results 1 – 11 of 11) sorted by relevance
29 auto shape_data = reinterpret_cast<const int *>(shape_tensor.Data().get()); in BuildLayer() local
39 int *shape_data = (int *)(shape->data_); in ScatterNdInferShape() local
179 auto shape_data = static_cast<int32_t *>(data); in GetShapeFromTensor() local183 auto shape_data = static_cast<int64_t *>(data); in GetShapeFromTensor() local
527 const auto *shape_data = reinterpret_cast<const int32_t *>(shape_tensor.GetData()); in GetShapeDataFromShapeTensor() local539 const auto *shape_data = reinterpret_cast<const int64_t *>(shape_tensor.GetData()); in GetShapeDataFromShapeTensor() local575 const int32_t *shape_data = reinterpret_cast<const int32_t *>(tensor.GetData()); in GetShapeDataFromConstData() local581 const int64_t *shape_data = reinterpret_cast<const int64_t *>(tensor.GetData()); in GetShapeDataFromConstData() local596 std::vector<int64_t> shape_data; in MakeShapeFromShapeTensor() local603 std::vector<int64_t> shape_data; in MakeShapeFromShapeTensor() local612 std::vector<int64_t> shape_data; in MakeDimForScalarInput() local
48 auto *shape_data = t.raw_data().data(); in Parse() local
55 std::vector<int64_t> shape_data; in Check() local
54 auto shape_data = reinterpret_cast<int *>(shape->data()); in ReSize() local
137 Tensor shape_data; in CUST_IMPLEMT_INFERFUNC() local461 Tensor shape_data; in CUST_IMPLEMT_INFERFUNC() local
235 T *shape_data = static_cast<T *>(shape_tensor->data_c()); in SetTensorInfoData() local
99 auto shape_data = std::vector<int64_t>({1, 1, 1, 1}); in GenExampleGraph() local
485 auto shape_data = std::vector<int64_t>({1, 1, 1, 1}); in GenExampleGraph() local