Searched refs:LiteralToHostTensor (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | literal_util_test.cc | 27 TEST(LiteralUtil, LiteralToHostTensor) { in TEST() argument
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()
62 EXPECT_TRUE(LiteralToHostTensor(int_values_literal, in TYPED_TEST()
68 EXPECT_TRUE(LiteralToHostTensor(int_values_literal, in TYPED_TEST()
78 LiteralToHostTensor(int_values_literal, DT_INT64, &host_tensor).code()); in TYPED_TEST()
|
| D | literal_util.h | 61 Status LiteralToHostTensor(const xla::LiteralSlice& literal,
|
| D | literal_util.cc | 101 Status LiteralToHostTensor(const xla::LiteralSlice& literal, in LiteralToHostTensor() function
|
| D | xla_expression.cc | 137 TF_RETURN_IF_ERROR(LiteralToHostTensor(literal, dtype(), &tensor)); in ResolveConstant()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 60 ctx, LiteralToHostTensor(begin_literal, index_type_, &begin_tensor)); in Compile()
62 LiteralToHostTensor(end_literal, index_type_, &end_tensor)); in Compile()
63 OP_REQUIRES_OK(ctx, LiteralToHostTensor(strides_literal, index_type_, in Compile()
144 ctx, LiteralToHostTensor(begin_literal, index_type_, &begin_tensor)); in Compile()
146 LiteralToHostTensor(end_literal, index_type_, &end_tensor)); in Compile()
147 OP_REQUIRES_OK(ctx, LiteralToHostTensor(strides_literal, index_type_, in Compile()
252 ctx, LiteralToHostTensor(begin_literal, index_type_, &begin_tensor)); in Compile()
254 LiteralToHostTensor(end_literal, index_type_, &end_tensor)); in Compile()
255 OP_REQUIRES_OK(ctx, LiteralToHostTensor(strides_literal, index_type_, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 250 TF_ASSERT_OK(LiteralToHostTensor(literal, DT_FLOAT, &tensor)); in TEST()
279 TF_ASSERT_OK(LiteralToHostTensor(literal0, DT_FLOAT, &tensor0)); in TEST()
281 TF_ASSERT_OK(LiteralToHostTensor(literal1, DT_FLOAT, &tensor1)); in TEST()
308 TF_ASSERT_OK(LiteralToHostTensor(literal0, DT_FLOAT, &tensor0)); in TEST()
333 TF_ASSERT_OK(LiteralToHostTensor(literal, DT_FLOAT, &tensor)); in TEST()
|