Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
H A Dresolve_svdf.cc186 tensorflow::TensorShapeProto* allocated_tensor_shape = in MaybeMergeConstNodes() local
188 auto tensor_shape_dim0 = allocated_tensor_shape->add_dim(); in MaybeMergeConstNodes()
204 allocated_tensor_shape->add_dim()->set_size(shape.dim(d).size()); in MaybeMergeConstNodes()
205 allocated_tensor_shape->set_unknown_rank(shape.unknown_rank()); in MaybeMergeConstNodes()
208 CHECK_EQ(shape.dim(d).size(), allocated_tensor_shape->dim(d).size()); in MaybeMergeConstNodes()
209 CHECK_EQ(allocated_tensor_shape->unknown_rank(), shape.unknown_rank()); in MaybeMergeConstNodes()
234 allocated_tensor_shape->clear_dim(); in MaybeMergeConstNodes()
235 allocated_tensor_shape->add_dim()->set_size(dim1_size); in MaybeMergeConstNodes()
236 allocated_tensor_shape->add_dim()->set_size(dim0_size); in MaybeMergeConstNodes()
H A Dresolve_svdf_test.cc106 tensorflow::TensorShapeProto* allocated_tensor_shape = in AddNewNode() local
108 auto tensor_shape_dim0 = allocated_tensor_shape->add_dim(); in AddNewNode()
110 allocated_tensor->set_allocated_tensor_shape(allocated_tensor_shape); in AddNewNode()
125 tensorflow::TensorShapeProto* allocated_tensor_shape = in AddShapeNode() local
127 auto tensor_shape_dim0 = allocated_tensor_shape->add_dim(); in AddShapeNode()
129 allocated_tensor->set_allocated_tensor_shape(allocated_tensor_shape); in AddShapeNode()