Home
last modified time | relevance | path

Searched refs:host_tensor (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dliteral_util.cc25 Status HostTensorToBorrowingLiteral(const Tensor& host_tensor, in HostTensorToBorrowingLiteral() argument
28 TF_RETURN_IF_ERROR(TensorShapeToXLAShape(host_tensor.dtype(), in HostTensorToBorrowingLiteral()
29 host_tensor.shape(), &xla_shape)); in HostTensorToBorrowingLiteral()
31 static_cast<const char*>(DMAHelper::base(&host_tensor)), xla_shape); in HostTensorToBorrowingLiteral()
35 xla::StatusOr<xla::Literal> HostTensorToLiteral(const Tensor& host_tensor) { in HostTensorToLiteral() argument
37 TF_RETURN_IF_ERROR(HostTensorToBorrowingLiteral(host_tensor, &literal)); in HostTensorToLiteral()
42 Tensor* host_tensor, xla::MutableBorrowingLiteral* literal) { in HostTensorToMutableBorrowingLiteral() argument
44 TF_RETURN_IF_ERROR(TensorShapeToXLAShape(host_tensor->dtype(), in HostTensorToMutableBorrowingLiteral()
45 host_tensor->shape(), &xla_shape)); in HostTensorToMutableBorrowingLiteral()
46 return HostTensorToMutableBorrowingLiteral(xla_shape, host_tensor, literal); in HostTensorToMutableBorrowingLiteral()
[all …]
Dliteral_util_test.cc32 Tensor host_tensor; in TEST() local
34 LiteralToHostTensor(int64_values_literal, DT_INT32, &host_tensor) in TEST()
37 LiteralToHostTensor(int64_values_literal, DT_QINT32, &host_tensor) in TEST()
40 LiteralToHostTensor(int64_values_literal, DT_INT64, &host_tensor).ok()); in TEST()
41 test::ExpectTensorEqual<int64>(host_tensor, in TEST()
58 Tensor host_tensor; in TYPED_TEST() local
63 DataTypeToEnum<int_type>::value, &host_tensor) in TYPED_TEST()
65 test::ExpectTensorEqual<int_type>(host_tensor, in TYPED_TEST()
70 &host_tensor) in TYPED_TEST()
73 test::ExpectTensorEqual<qint_type>(host_tensor, in TYPED_TEST()
[all …]
Dliteral_util.h31 Status HostTensorToBorrowingLiteral(const Tensor& host_tensor,
36 xla::StatusOr<xla::Literal> HostTensorToLiteral(const Tensor& host_tensor);
41 Tensor* host_tensor, xla::MutableBorrowingLiteral* literal);
46 const xla::Shape& xla_shape, Tensor* host_tensor,
62 DataType target_type, Tensor* host_tensor);
68 Tensor* host_tensor);
Dxla_op_kernel.h175 void SetConstantOutput(int index, const Tensor& host_tensor);
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compile_on_demand_op.cc126 constant_arguments[i] = xla_tensor->host_tensor(); in Compile()
139 Tensor host_tensor; in Compile() local
143 device_tensor.dtype(), device_tensor.shape(), &host_tensor, attrs)); in Compile()
148 reinterpret_cast<Device*>(ctx->device()), &host_tensor, in Compile()
161 constant_arguments[i] = host_tensor; in Compile()
Dxla_tensor.h85 const Tensor& host_tensor() const { return *host_tensor_; } in host_tensor() function
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_test.cc2013 Tensor host_tensor(dtype, {3}); in TestFeedAndFetchTensorsInDeviceMemory() local
2023 TF_ASSERT_OK(session->RunCallable(handle, {host_tensor}, &outputs, nullptr)) in TestFeedAndFetchTensorsInDeviceMemory()
2044 const StringPiece expected_data = host_tensor.tensor_data(); in TestFeedAndFetchTensorsInDeviceMemory()