| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_creation_utils_test.cc | 60 Literal result_literal,
62 CHECK_EQ(result_literal, LiteralUtil::CreateR1<int32>({3, 4}));
79 Literal result_literal,
83 CHECK_EQ(result_literal,
102 Literal result_literal,
104 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32>({{9, 10}}));
121 Literal result_literal,
123 CHECK_EQ(result_literal, LiteralUtil::CreateR3<int32>({{{9, 10}}}));
140 Literal result_literal,
142 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32>({{9}}))
[all...] |
| service.cc | [all...] |
| hlo_evaluator.cc | 473 auto result_literal = LiteralUtil::CreateFromDimensions( local
480 TF_RETURN_IF_ERROR(result_literal.CopySliceFrom(
487 evaluated_[concatenate] = std::move(result_literal);
[all...] |
| hlo_evaluator_test.cc | 1760 Literal result_literal = local
[all...] |
| hlo_evaluator_typed_visitor.h | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| executable.cc | 97 Literal result_literal; local
101 TF_ASSIGN_OR_RETURN(result_literal,
108 result_literal.shape(), run_options->allocator(),
111 run_options->stream(), result_literal, result));
|
| /external/tensorflow/tensorflow/compiler/xla/rpc/ |
| grpc_client_test.cc | 100 TF_ASSERT_OK_AND_ASSIGN(auto result_literal, client_->ExecuteAndTransfer(
102 EXPECT_TRUE(LiteralTestUtil::Near(expected_literal, result_literal,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| local_client_execute_test.cc | 211 Literal result_literal = ShapedBufferToLiteral(result); local
213 LiteralSlice(result_literal, {0}));
215 LiteralSlice(result_literal, {1}));
217 LiteralSlice(result_literal, {2}));
239 Literal result_literal = ShapedBufferToLiteral(result); local
241 LiteralSlice(result_literal, {1}));
243 LiteralSlice(result_literal, {0, 0}));
245 LiteralSlice(result_literal, {0, 1}));
247 LiteralSlice(result_literal, {0, 2}));
271 Literal result_literal = ShapedBufferToLiteral(result) local
317 Literal result_literal = ShapedBufferToLiteral(result); local
356 Literal result_literal = ShapedBufferToLiteral(result); local
437 Literal result_literal = ShapedBufferToLiteral(result); local
491 Literal result_literal = ShapedBufferToLiteral(result); local
535 Literal result_literal = ShapedBufferToLiteral(result); local
[all...] |
| client_test.cc | 146 auto result_literal,
149 EXPECT_TRUE(LiteralTestUtil::Equal(expected_result, result_literal));
|
| exhaustive_op_test.cc | 273 TF_ASSERT_OK_AND_ASSIGN(Literal result_literal,
275 ExpectNear<T>(input_literal, result_literal, evaluate_op);
310 TF_ASSIGN_OR_RETURN(Literal result_literal,
312 return std::move(result_literal);
332 void ExpectNear(const Literal& input_literal, const Literal& result_literal,
343 absl::Span<const T> result_arr = result_literal.data<T>();
|
| gather_operation_test.cc | 674 TF_ASSERT_OK_AND_ASSIGN(Literal result_literal,
676 LiteralTestUtil::ExpectR2Equal<int32>({{1, 2, 3}, {7, 8, 9}}, result_literal);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| cpu_infeed_test.cc | 171 auto result_literal = client_->Transfer(*result).ConsumeValueOrDie(); local
174 LiteralTestUtil::ExpectR0Near<float>(45.0f, result_literal, ErrorSpec{1e-7});
289 auto result_literal = client_->Transfer(*result).ConsumeValueOrDie(); local
292 LiteralTestUtil::ExpectR0Near<float>(66.0f, result_literal, ErrorSpec{1e-7});
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| reference_util.cc | 553 Literal result_literal = local
556 CHECK_EQ(result_literal.shape().rank(), 4);
558 absl::make_unique<Array4D<float>>(result_literal.shape().dimensions(0),
559 result_literal.shape().dimensions(1),
560 result_literal.shape().dimensions(2),
561 result_literal.shape().dimensions(3));
564 *value = result_literal.Get<float>(indices);
|
| literal.cc | 716 Literal result_literal(result_shape);
718 result_literal.EachCell<NativeT>(
724 result_literal.Set<NativeT>(indices, value);
726 return result_literal;
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| replay_computation.cc | 304 TF_ASSIGN_OR_RETURN(Literal result_literal,
306 return result_literal;
|
