Home
last modified time | relevance | path

Searched refs:allocated_tensor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
H A Dresolve_svdf_test.cc105 tensorflow::TensorProto* allocated_tensor = new tensorflow::TensorProto; in AddNewNode() local
110 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() local
129 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()
H A Dresolve_svdf.cc184 tensorflow::TensorProto* allocated_tensor = in MaybeMergeConstNodes() local
187 allocated_tensor->mutable_tensor_shape(); in MaybeMergeConstNodes()
194 allocated_tensor->set_dtype(tensor.dtype()); in MaybeMergeConstNodes()
196 CHECK_EQ(allocated_tensor->dtype(), tensor.dtype()); in MaybeMergeConstNodes()
239 allocated_tensor->set_tensor_content( in MaybeMergeConstNodes()
246 allocated_tensor->set_tensor_content( in MaybeMergeConstNodes()