Searched refs:LiteralToHostTensor (Results 1 - 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | literal_util_test.cc | 25 TEST(LiteralUtil, LiteralToHostTensor) {
33 LiteralToHostTensor(*int64_values_literal, DT_INT32, &host_tensor)
37 LiteralToHostTensor(*int64_values_literal, DT_QINT32, &host_tensor)
40 LiteralToHostTensor(*int64_values_literal, DT_INT64, &host_tensor)
53 LiteralToHostTensor(*int32_values_literal, DT_INT32, &host_tensor)
59 LiteralToHostTensor(*int32_values_literal, DT_QINT32, &host_tensor)
66 LiteralToHostTensor(*int32_values_literal, DT_INT64, &host_tensor)
|
| H A D | literal_util.h | 39 Status LiteralToHostTensor(const xla::Literal& literal, DataType target_type,
|
| H A D | literal_util.cc | 66 Status LiteralToHostTensor(const xla::Literal& literal, DataType target_type, function in namespace:tensorflow
|
| H A D | graph_compiler.cc | 78 LiteralToHostTensor(*literal, arg.type, &arg.constant_value));
|
| H A D | xla_context.cc | 99 TF_RETURN_IF_ERROR(LiteralToHostTensor(literal, dtype, &value));
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | strided_slice_op.cc | 59 ctx, LiteralToHostTensor(begin_literal, index_type_, &begin_tensor));
61 LiteralToHostTensor(end_literal, index_type_, &end_tensor));
62 OP_REQUIRES_OK(ctx, LiteralToHostTensor(strides_literal, index_type_,
142 ctx, LiteralToHostTensor(begin_literal, index_type_, &begin_tensor));
144 LiteralToHostTensor(end_literal, index_type_, &end_tensor));
145 OP_REQUIRES_OK(ctx, LiteralToHostTensor(strides_literal, index_type_,
250 ctx, LiteralToHostTensor(begin_literal, index_type_, &begin_tensor));
252 LiteralToHostTensor(end_literal, index_type_, &end_tensor));
253 OP_REQUIRES_OK(ctx, LiteralToHostTensor(strides_literal, index_type_,
|
Completed in 265 milliseconds