Searched refs:allocated_tensor (Results 1 – 3 of 3) sorted by relevance
107 TfLiteTensor allocated_tensor; in TF_LITE_MICRO_TEST() local111 &allocated_tensor)); in TF_LITE_MICRO_TEST()112 TF_LITE_MICRO_EXPECT_EQ(kTfLiteInt32, allocated_tensor.type); in TF_LITE_MICRO_TEST()113 TF_LITE_MICRO_EXPECT_EQ(1, allocated_tensor.dims->size); in TF_LITE_MICRO_TEST()114 TF_LITE_MICRO_EXPECT_EQ(100, allocated_tensor.dims->data[0]); in TF_LITE_MICRO_TEST()115 TF_LITE_MICRO_EXPECT_EQ(400, allocated_tensor.bytes); in TF_LITE_MICRO_TEST()116 TF_LITE_MICRO_EXPECT_NE(nullptr, allocated_tensor.data.i32); in TF_LITE_MICRO_TEST()128 TfLiteTensor allocated_tensor; in TF_LITE_MICRO_TEST() local132 &allocated_tensor)); in TF_LITE_MICRO_TEST()
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()