Searched refs:tensor_shape_proto (Results 1 – 7 of 7) sorted by relevance
75 TensorShapeProto& tensor_shape_proto = *tensor_shape_type.mutable_shape(); in BuildRemoteFusedGraphExecuteInfo() local77 tensor_shape_proto.add_dim()->set_size(dim); in BuildRemoteFusedGraphExecuteInfo()90 TensorShapeProto& tensor_shape_proto = in BuildRemoteFusedGraphExecuteInfo() local93 tensor_shape_proto.add_dim()->set_size(dim); in BuildRemoteFusedGraphExecuteInfo()
220 TensorShapeProto& tensor_shape_proto = *tensor_shape_type.mutable_shape(); in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo() local222 tensor_shape_proto.add_dim()->set_size(dim); in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo()232 TensorShapeProto& tensor_shape_proto = *tensor_shape_type.mutable_shape(); in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo() local234 tensor_shape_proto.add_dim()->set_size(dim); in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo()
926 const TensorShapeProto& tensor_shape_proto) { in ExpectTensorShape() argument928 TensorShape tensor_shape(tensor_shape_proto); in ExpectTensorShape()937 TensorShapeProto tensor_shape_proto; in TEST_F() local943 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()950 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()955 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()963 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()964 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()972 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()973 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()[all …]
28 TensorShapeProto* tensor_shape_proto);
866 TensorShapeProto* tensor_shape_proto) { in GetTensorShapeProtoFromTensorProto() argument867 tensor_shape_proto->Clear(); in GetTensorShapeProtoFromTensorProto()892 tensor_shape_proto->add_dim()->set_size(value); \ in GetTensorShapeProtoFromTensorProto()
280 TensorShapeProto tensor_shape_proto;281 internal::SetTensorProtoShape(shape, &tensor_shape_proto);282 if (TensorShape(tensor_shape_proto).num_elements() != values.size()) {289 tensor.mutable_tensor_shape()->Swap(&tensor_shape_proto);
632 TensorShapeProto& tensor_shape_proto = *tensor_shape_type.mutable_shape(); in BuildRemoteFusedGraphExecuteInfo() local634 tensor_shape_proto.add_dim()->set_size(dim); in BuildRemoteFusedGraphExecuteInfo()657 TensorShapeProto& tensor_shape_proto = in BuildRemoteFusedGraphExecuteInfo() local660 tensor_shape_proto.add_dim()->set_size(dim); in BuildRemoteFusedGraphExecuteInfo()