Home
last modified time | relevance | path

Searched refs:result_literal (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils_test.cc75 Literal result_literal, in TEST_F()
77 CHECK_EQ(result_literal, LiteralUtil::CreateR1<int32>({3, 4})); in TEST_F()
94 Literal result_literal, in TEST_F()
98 CHECK_EQ(result_literal, in TEST_F()
117 Literal result_literal, in TEST_F()
119 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32>({{9, 10}})); in TEST_F()
136 Literal result_literal, in TEST_F()
138 CHECK_EQ(result_literal, LiteralUtil::CreateR3<int32>({{{9, 10}}})); in TEST_F()
155 Literal result_literal, in TEST_F()
157 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32>({{9}})); in TEST_F()
[all …]
Dservice.cc935 Literal result_literal, in TransferToClient()
939 if (LayoutUtil::LayoutsInShapesEqual(return_shape, result_literal.shape())) { in TransferToClient()
940 *result->mutable_literal() = result_literal.ToProto(); in TransferToClient()
943 result_literal.Relayout(return_shape).ToProto(); in TransferToClient()
1082 TF_ASSIGN_OR_RETURN(auto result_literal, evaluator.Evaluate(*module, {})); in ComputeConstantGraph()
1089 result_literal = result_literal.Relayout(*output_layout); in ComputeConstantGraph()
1091 *result->mutable_literal() = result_literal.ToProto(); in ComputeConstantGraph()
Dhlo_evaluator.cc510 auto result_literal = LiteralUtil::CreateFromDimensions( in HandleConcatenate() local
517 TF_RETURN_IF_ERROR(result_literal.CopySliceFrom( in HandleConcatenate()
524 evaluated_[concatenate] = std::move(result_literal); in HandleConcatenate()
Dhlo_evaluator_typed_visitor.h2963 auto result_literal,
2967 return std::move(result_literal);
Dhlo_evaluator_test.cc2860 Literal result_literal = in TEST_P() local
2862 EXPECT_TRUE(LiteralTestUtil::Equal(result_literal, result)); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable_base.cc123 TF_ASSIGN_OR_RETURN(Literal result_literal, in ExecuteAsyncOnStream()
126 result_literal = result_literal.ToStatic(); in ExecuteAsyncOnStream()
131 result_literal.shape(), run_options->allocator(), in ExecuteAsyncOnStream()
134 run_options->stream(), result_literal, result_buffers)); in ExecuteAsyncOnStream()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_execute_test.cc214 Literal result_literal = ShapedBufferToLiteral(result); in XLA_TEST_F() local
216 LiteralSlice(result_literal, {0})); in XLA_TEST_F()
218 LiteralSlice(result_literal, {1})); in XLA_TEST_F()
220 LiteralSlice(result_literal, {2})); in XLA_TEST_F()
242 Literal result_literal = ShapedBufferToLiteral(result); in XLA_TEST_F() local
244 LiteralSlice(result_literal, {1})); in XLA_TEST_F()
246 LiteralSlice(result_literal, {0, 0})); in XLA_TEST_F()
248 LiteralSlice(result_literal, {0, 1})); in XLA_TEST_F()
250 LiteralSlice(result_literal, {0, 2})); in XLA_TEST_F()
274 Literal result_literal = ShapedBufferToLiteral(result); in XLA_TEST_F() local
[all …]
Dclient_test.cc146 auto result_literal, in XLA_TEST_F()
149 EXPECT_TRUE(LiteralTestUtil::Equal(expected_result, result_literal)); in XLA_TEST_F()
Dexhaustive_op_test_utils.h203 TF_ASSERT_OK_AND_ASSIGN(Literal result_literal,
205 ExpectNear(input_literals, result_literal, evaluate_op, error_spec_gen);
231 const Literal& result_literal, EvaluateOp evaluate_op,
276 TF_ASSIGN_OR_RETURN(Literal result_literal,
278 return std::move(result_literal);
Dbuffer_donation_test.cc160 Literal result_literal, in RunAndCheck()
163 EXPECT_TRUE(LiteralTestUtil::Equal(expected, result_literal)); in RunAndCheck()
Dgather_operation_test.cc785 TF_ASSERT_OK_AND_ASSIGN(Literal result_literal, in XLA_TEST_F()
787 LiteralTestUtil::ExpectR2Equal<int32>({{1, 2, 3}, {7, 8, 9}}, result_literal); in XLA_TEST_F()
Dexhaustive_op_test_utils.cc325 const Literal& result_literal, in ExpectNear() argument
349 absl::Span<const NativeT> result_arr = result_literal.data<NativeT>(); in ExpectNear()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_infeed_test.cc171 auto result_literal = client_->Transfer(*result).ConsumeValueOrDie(); in TEST_F() local
174 LiteralTestUtil::ExpectR0Near<float>(45.0f, result_literal, ErrorSpec{1e-7}); in TEST_F()
289 auto result_literal = client_->Transfer(*result).ConsumeValueOrDie(); in TEST_F() local
292 LiteralTestUtil::ExpectR0Near<float>(66.0f, result_literal, ErrorSpec{1e-7}); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/rpc/
Dgrpc_client_test.cc106 TF_ASSERT_OK_AND_ASSIGN(auto result_literal, client_->ExecuteAndTransfer( in TEST_F()
108 EXPECT_TRUE(LiteralTestUtil::Near(expected_literal, result_literal, in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc488 Literal result_literal = in ConvArray4DGeneralDimensionsDilated() local
491 CHECK_EQ(result_literal.shape().rank(), 4); in ConvArray4DGeneralDimensionsDilated()
493 absl::make_unique<Array4D<float>>(result_literal.shape().dimensions(0), in ConvArray4DGeneralDimensionsDilated()
494 result_literal.shape().dimensions(1), in ConvArray4DGeneralDimensionsDilated()
495 result_literal.shape().dimensions(2), in ConvArray4DGeneralDimensionsDilated()
496 result_literal.shape().dimensions(3)); in ConvArray4DGeneralDimensionsDilated()
499 *value = result_literal.Get<float>(indices); in ConvArray4DGeneralDimensionsDilated()
Dliteral.cc891 Literal result_literal(result_shape); in SliceInternal() local
893 CHECK(result_literal in SliceInternal()
906 result_literal.SetDynamicSize(dnum, dynamic_size); in SliceInternal()
909 return result_literal; in SliceInternal()
1343 Literal result_literal(ShapeUtil::ChangeElementType( in ConvertBetweenNativeTypesWithConverter() local
1347 auto dest_data = result_literal.template data<NativeDestT>(); in ConvertBetweenNativeTypesWithConverter()
1353 return result_literal; in ConvertBetweenNativeTypesWithConverter()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dreplay_computation.cc345 TF_ASSIGN_OR_RETURN(Literal result_literal, in ReplayComputation()
347 return result_literal; in ReplayComputation()