| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils_test.cc | 77 Literal result_literal, in TEST_F()
79 CHECK_EQ(result_literal, LiteralUtil::CreateR1<int32_t>({3, 4})); in TEST_F()
96 Literal result_literal, in TEST_F()
100 CHECK_EQ(result_literal, in TEST_F()
119 Literal result_literal, in TEST_F()
121 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32_t>({{9, 10}})); in TEST_F()
138 Literal result_literal, in TEST_F()
140 CHECK_EQ(result_literal, LiteralUtil::CreateR3<int32_t>({{{9, 10}}})); in TEST_F()
157 Literal result_literal, in TEST_F()
159 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32_t>({{9}})); in TEST_F()
[all …]
|
| D | service.cc | 998 Literal result_literal, in TransferToClient()
1002 if (LayoutUtil::LayoutsInShapesEqual(return_shape, result_literal.shape())) { in TransferToClient()
1003 *result->mutable_literal() = result_literal.ToProto(); in TransferToClient()
1006 result_literal.Relayout(return_shape).ToProto(); in TransferToClient()
1166 TF_ASSIGN_OR_RETURN(auto result_literal, evaluator.Evaluate(*module, {})); in ComputeConstantGraph()
1173 result_literal = result_literal.Relayout(*output_layout); in ComputeConstantGraph()
1175 *result->mutable_literal() = result_literal.ToProto(); in ComputeConstantGraph()
|
| D | hlo_evaluator.cc | 1164 auto result_literal = LiteralUtil::CreateFromDimensions( in HandleConcatenate() local
1171 TF_RETURN_IF_ERROR(result_literal.CopySliceFrom( in HandleConcatenate()
1178 evaluated_[concatenate] = std::move(result_literal); in HandleConcatenate()
|
| D | hlo_evaluator_typed_visitor.h | 2422 auto result_literal,
2426 return std::move(result_literal);
|
| D | hlo_evaluator_test.cc | 2898 Literal result_literal = in TEST_P() local
2900 EXPECT_TRUE(LiteralTestUtil::Equal(result_literal, result)); in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_execute_test.cc | 214 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 …]
|
| D | client_test.cc | 146 auto result_literal, in XLA_TEST_F()
149 EXPECT_TRUE(LiteralTestUtil::Equal(expected_result, result_literal)); in XLA_TEST_F()
|
| D | exhaustive_op_test_utils.h | 209 TF_ASSERT_OK_AND_ASSIGN(Literal result_literal,
211 ExpectNear(input_literals, result_literal, evaluate_op, error_spec_gen);
237 const Literal& result_literal, EvaluateOp evaluate_op,
282 TF_ASSIGN_OR_RETURN(Literal result_literal,
284 return std::move(result_literal);
|
| D | buffer_donation_test.cc | 163 Literal result_literal, in RunAndCheck()
166 EXPECT_TRUE(LiteralTestUtil::Equal(expected, result_literal)); in RunAndCheck()
|
| D | gather_operation_test.cc | 785 TF_ASSERT_OK_AND_ASSIGN(Literal result_literal, in XLA_TEST_F()
788 result_literal); in XLA_TEST_F()
|
| D | exhaustive_op_test_utils.cc | 325 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/interpreter/ |
| D | executable_base.cc | 125 TF_ASSIGN_OR_RETURN(Literal result_literal, in ExecuteAsyncOnStream()
128 result_literal = result_literal.ToStatic(); in ExecuteAsyncOnStream()
136 result_literal.shape(), alias_config, in ExecuteAsyncOnStream()
140 run_options->stream(), result_literal, result.Result())); in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_infeed_test.cc | 171 auto result_literal = client_->Transfer(*result).value(); 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).value(); in TEST_F() local
292 LiteralTestUtil::ExpectR0Near<float>(66.0f, result_literal, ErrorSpec{1e-7}); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/rpc/ |
| D | grpc_client_test.cc | 105 TF_ASSERT_OK_AND_ASSIGN(auto result_literal, client_->ExecuteAndTransfer( in TEST_F()
107 EXPECT_TRUE(LiteralTestUtil::Near(expected_literal, result_literal, in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 491 Literal result_literal = evaluator.Evaluate(*computation, {}).value(); in ConvArray4DGeneralDimensionsDilated() local
493 CHECK_EQ(result_literal.shape().rank(), 4); in ConvArray4DGeneralDimensionsDilated()
495 std::make_unique<Array4D<float>>(result_literal.shape().dimensions(0), in ConvArray4DGeneralDimensionsDilated()
496 result_literal.shape().dimensions(1), in ConvArray4DGeneralDimensionsDilated()
497 result_literal.shape().dimensions(2), in ConvArray4DGeneralDimensionsDilated()
498 result_literal.shape().dimensions(3)); in ConvArray4DGeneralDimensionsDilated()
501 *value = result_literal.Get<float>(indices); in ConvArray4DGeneralDimensionsDilated()
|
| D | literal.cc | 1031 Literal result_literal(result_shape); in SliceInternal() local
1033 CHECK(result_literal in SliceInternal()
1046 result_literal.SetDynamicSize(dnum, dynamic_size); in SliceInternal()
1049 return result_literal; in SliceInternal()
1494 Literal result_literal(ShapeUtil::ChangeElementType( in ConvertBetweenNativeTypesWithConverter() local
1498 auto dest_data = result_literal.template data<NativeDestT>(); in ConvertBetweenNativeTypesWithConverter()
1504 return result_literal; in ConvertBetweenNativeTypesWithConverter()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 348 TF_ASSIGN_OR_RETURN(Literal result_literal, in ReplayComputation()
350 return result_literal; in ReplayComputation()
|