Searched refs:allocated_tensor_shape (Results 1 – 2 of 2) sorted by relevance
185 tensorflow::TensorShapeProto* allocated_tensor_shape = in MaybeMergeConstNodes() local187 auto tensor_shape_dim0 = allocated_tensor_shape->add_dim(); in MaybeMergeConstNodes()203 allocated_tensor_shape->add_dim()->set_size(shape.dim(d).size()); in MaybeMergeConstNodes()204 allocated_tensor_shape->set_unknown_rank(shape.unknown_rank()); in MaybeMergeConstNodes()207 CHECK_EQ(shape.dim(d).size(), allocated_tensor_shape->dim(d).size()); in MaybeMergeConstNodes()208 CHECK_EQ(allocated_tensor_shape->unknown_rank(), shape.unknown_rank()); in MaybeMergeConstNodes()233 allocated_tensor_shape->clear_dim(); in MaybeMergeConstNodes()234 allocated_tensor_shape->add_dim()->set_size(dim1_size); in MaybeMergeConstNodes()235 allocated_tensor_shape->add_dim()->set_size(dim0_size); in MaybeMergeConstNodes()
106 tensorflow::TensorShapeProto* allocated_tensor_shape = in AddNewNode() local108 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() local127 auto tensor_shape_dim0 = allocated_tensor_shape->add_dim(); in AddShapeNode()129 allocated_tensor->set_allocated_tensor_shape(allocated_tensor_shape); in AddShapeNode()