Searched defs:host_tensor (Results 1 – 8 of 8) sorted by relevance
25 Status HostTensorToBorrowingLiteral(const Tensor& host_tensor, in HostTensorToBorrowingLiteral()34 const Tensor& host_tensor, in HostTensorToBorrowingLiteral()46 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
180 const auto& host_tensor = tensor_handle.GetAsyncTensor()->get<HostTensor>(); in CreateTFTensorFromTensorHandle() local
377 const auto& host_tensor = (dst_device == kCPU ? dst_tensor : src_tensor); in copy_kernel_cuda() local
255 const auto& host_tensor = result->get<FallbackTensor>().tensor(); in GraphExecutionRunOnFunction() local
942 Tensor* host_tensor) { in CopyTensorToHost()
590 Tensor host_tensor(gpu_tensor.dtype(), gpu_tensor.shape()); in CopyGpuTensorToHostDebugOnly() local
2475 Tensor host_tensor(dtype, {3}); in TestFeedAndFetchTensorsInDeviceMemory() local