Searched refs:AsProtoTensorContent (Results 1 – 25 of 47) sorted by relevance
12
48 tensor.AsProtoTensorContent(v->mutable_tensor()); in Compute()90 tensor.AsProtoTensorContent(v->mutable_tensor()); in Compute()
80 tensor.AsProtoTensorContent(&proto); in Compute()
116 input.AsProtoTensorContent(&proto); in Serialize()158 input.AsProtoTensorContent(&proto); in operator ()()
66 feeds_[i].second.AsProtoTensorContent(out_tensor); in FeedValue()119 feeds_[i].second.AsProtoTensorContent(feed->mutable_tensor()); in ToProto()172 value.AsProtoTensorContent(value_proto); in add_feed()411 sends_[i].second.AsProtoTensorContent(send->mutable_tensor()); in ToProto()622 recvs_[i].second.AsProtoTensorContent(out_tensor); in RecvValue()700 value.AsProtoTensorContent(value_proto); in AddRecv()771 value.AsProtoTensorContent(value_proto); in AddRecv()
82 src.AsProtoTensorContent(proto.mutable_tensor()); in Validate()170 src.AsProtoTensorContent(proto.mutable_tensor()); in MakeFloatTensorTestCase()
284 args[i].AsProtoTensorContent(send->mutable_tensor()); in Run()
58 tensor.AsProtoTensorContent(&p); in TensorProtoHash()89 lhs_t.AsProtoTensorContent(&lhs_tp); in AreTensorProtosEqual()92 rhs_t.AsProtoTensorContent(&rhs_tp); in AreTensorProtosEqual()549 value.AsProtoTensorContent(out->mutable_tensor()); in SetAttrValue()559 v.AsProtoTensorContent(out->mutable_list()->add_tensor()); in SetAttrValue()
33 // Serialized raw tensor content from either Tensor::AsProtoTensorContent or
47 original.AsProtoTensorContent(&proto); in TEST()
362 paddings_t.AsProtoTensorContent(permuted_paddings_tensor.mutable_tensor()); in EraseCancellableNodesAroundPad()
430 tensor.AsProtoTensorContent(attr_tensor.mutable_tensor()); in AddNodePermConst()454 tensor.AsProtoTensorContent(attr_tensor.mutable_tensor()); in AddNodeConstScalar()859 tensor.AsProtoTensorContent( in UpdateAttrValue()1415 tensor.AsProtoTensorContent(attr_tensor.mutable_tensor()); in AddNodeShapeConst()
71 tensors[j].AsProtoTensorContent(&proto); in SnapshotRoundTrip()72 read_tensors[j].AsProtoTensorContent(&read_proto); in SnapshotRoundTrip()
119 tensor.AsProtoTensorContent(&proto); in WriteTensors()206 tensor.AsProtoTensorContent(t); in WriteTensors()238 tensor.AsProtoTensorContent(&proto); in WriteTensors()
67 expected.AsProtoTensorContent(&proto); in TEST_P()
37 component.AsProtoTensorContent(&non_memcpy_components.back()); in CompressElement()
263 shp_tensor.AsProtoTensorContent( in RunInfiniteTFLoop()295 true_tensor.AsProtoTensorContent( in RunInfiniteTFLoop()403 tensor.AsProtoTensorContent(&tensor_proto); in SetNodeAttr()
162 val.AsProtoTensorContent(response.mutable_tensor()); in EncodeTensorToByteBuffer()
91 parsed.AsProtoTensorContent(proto); in ReEncodeConsts()449 feed.AsProtoTensorContent(req.mutable_feed()->Add()); in RunCallable()
84 t.AsProtoTensorContent(v->mutable_tensor()); in WriteTensor()
326 tensor.AsProtoTensorContent(local_handle->mutable_tensor()); in SerializePackedHandle()437 tensor.AsProtoTensorContent(send_tensor->add_tensors()); in StartRemoteSendTensor()
181 absl::get<Tensor>(arg).AsProtoTensorContent( in Run()
87 tensor.AsProtoTensorContent(&tensor_proto); in SetNodeTensorAttr()
161 variable_value.AsProtoTensorContent( in ConvertVariableToConstant()
185 tensor.AsProtoTensorContent(&tensor_proto);
71 input_tensor.AsProtoTensorContent(&tensor_proto); in ConvertToProto()