Searched defs:host_tensor (Results 1 – 5 of 5) sorted by relevance
25 Status HostTensorToBorrowingLiteral(const Tensor& host_tensor, in HostTensorToBorrowingLiteral()35 xla::StatusOr<xla::Literal> HostTensorToLiteral(const Tensor& host_tensor) { in HostTensorToLiteral()42 Tensor* host_tensor, xla::MutableBorrowingLiteral* literal) { in HostTensorToMutableBorrowingLiteral()50 const xla::Shape& xla_shape, Tensor* host_tensor, in HostTensorToMutableBorrowingLiteral()79 Tensor* host_tensor) { in CopyLiteralToHostTensor()102 DataType target_type, Tensor* host_tensor) { in LiteralToHostTensor()
32 Tensor host_tensor; in TEST() local58 Tensor host_tensor; in TYPED_TEST() local
85 const Tensor& host_tensor() const { return *host_tensor_; } in host_tensor() function
139 Tensor host_tensor; in Compile() local
2013 Tensor host_tensor(dtype, {3}); in TestFeedAndFetchTensorsInDeviceMemory() local