Searched refs:host_tensors (Results 1 – 2 of 2) sorted by relevance
58 Status HostTensorsToBorrowingLiteralTuple(absl::Span<const Tensor> host_tensors, in HostTensorsToBorrowingLiteralTuple() argument61 buf_ptrs.reserve(host_tensors.size()); in HostTensorsToBorrowingLiteralTuple()62 std::vector<xla::Shape> tensor_shapes(host_tensors.size()); in HostTensorsToBorrowingLiteralTuple()64 for (int i = 0; i < host_tensors.size(); i++) { in HostTensorsToBorrowingLiteralTuple()66 const Tensor* tensor = &host_tensors[i]; in HostTensorsToBorrowingLiteralTuple()
51 Status HostTensorsToBorrowingLiteralTuple(absl::Span<const Tensor> host_tensors,