Searched defs:host_tensor (Results 1 – 3 of 3) sorted by relevance
25 Status HostTensorToBorrowingLiteral(const Tensor& host_tensor, in HostTensorToBorrowingLiteral()34 const Tensor& host_tensor, in HostTensorToBorrowingLiteral()46 xla::StatusOr<xla::Literal> HostTensorToLiteral(const Tensor& host_tensor) { in HostTensorToLiteral()53 Tensor* host_tensor, xla::MutableBorrowingLiteral* literal) { in HostTensorToMutableBorrowingLiteral()61 const xla::Shape& xla_shape, Tensor* host_tensor, in HostTensorToMutableBorrowingLiteral()90 Tensor* host_tensor) { in CopyLiteralToHostTensor()113 DataType target_type, Tensor* host_tensor) { in LiteralToHostTensor()
32 Tensor host_tensor; in TEST() local58 Tensor host_tensor; in TYPED_TEST() local
2442 Tensor host_tensor(dtype, {3}); in TestFeedAndFetchTensorsInDeviceMemory() local