Searched refs:allocated_tensor (Results 1 – 3 of 3) sorted by relevance
105 tensorflow::TensorProto* allocated_tensor = new tensorflow::TensorProto; in AddNewNode() local110 allocated_tensor->set_allocated_tensor_shape(allocated_tensor_shape); in AddNewNode()111 allocated_tensor->set_tensor_content( in AddNewNode()114 (*node->mutable_attr())["value"].set_allocated_tensor(allocated_tensor); in AddNewNode()124 tensorflow::TensorProto* allocated_tensor = new tensorflow::TensorProto; in AddShapeNode() local129 allocated_tensor->set_allocated_tensor_shape(allocated_tensor_shape); in AddShapeNode()130 allocated_tensor->set_tensor_content( in AddShapeNode()133 (*node->mutable_attr())["value"].set_allocated_tensor(allocated_tensor); in AddShapeNode()
183 tensorflow::TensorProto* allocated_tensor = in MaybeMergeConstNodes() local186 allocated_tensor->mutable_tensor_shape(); in MaybeMergeConstNodes()193 allocated_tensor->set_dtype(tensor.dtype()); in MaybeMergeConstNodes()195 CHECK_EQ(allocated_tensor->dtype(), tensor.dtype()); in MaybeMergeConstNodes()238 allocated_tensor->set_tensor_content( in MaybeMergeConstNodes()245 allocated_tensor->set_tensor_content( in MaybeMergeConstNodes()
137 TfLiteTensor allocated_tensor; in TF_LITE_MICRO_TEST() local142 tflite::GetMicroErrorReporter(), &allocated_tensor)); in TF_LITE_MICRO_TEST()143 TF_LITE_MICRO_EXPECT_EQ(kTfLiteInt32, allocated_tensor.type); in TF_LITE_MICRO_TEST()144 TF_LITE_MICRO_EXPECT_EQ(1, allocated_tensor.dims->size); in TF_LITE_MICRO_TEST()145 TF_LITE_MICRO_EXPECT_EQ(100, allocated_tensor.dims->data[0]); in TF_LITE_MICRO_TEST()146 TF_LITE_MICRO_EXPECT_EQ(static_cast<size_t>(400), allocated_tensor.bytes); in TF_LITE_MICRO_TEST()147 TF_LITE_MICRO_EXPECT(nullptr == allocated_tensor.data.i32); in TF_LITE_MICRO_TEST()148 TF_LITE_MICRO_EXPECT_EQ(kTfLiteArenaRw, allocated_tensor.allocation_type); in TF_LITE_MICRO_TEST()196 TfLiteTensor allocated_tensor; in TF_LITE_MICRO_TEST() local201 tflite::GetMicroErrorReporter(), &allocated_tensor)); in TF_LITE_MICRO_TEST()[all …]