| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils_test.cc | 60 Literal result_literal, in TEST_F()
62 CHECK_EQ(result_literal, LiteralUtil::CreateR1<int32>({3, 4})); in TEST_F()
79 Literal result_literal, in TEST_F()
83 CHECK_EQ(result_literal, in TEST_F()
102 Literal result_literal, in TEST_F()
104 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32>({{9, 10}})); in TEST_F()
121 Literal result_literal, in TEST_F()
123 CHECK_EQ(result_literal, LiteralUtil::CreateR3<int32>({{{9, 10}}})); in TEST_F()
140 Literal result_literal, in TEST_F()
142 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32>({{9}})); in TEST_F()
[all …]
|
| D | service.cc | 958 Literal result_literal, in TransferToClient()
962 if (LayoutUtil::LayoutsInShapesEqual(return_shape, result_literal.shape())) { in TransferToClient()
963 *result->mutable_literal() = result_literal.ToProto(); in TransferToClient()
966 result_literal.Relayout(return_shape).ToProto(); in TransferToClient()
1105 TF_ASSIGN_OR_RETURN(auto result_literal, evaluator.Evaluate(*module, {})); in ComputeConstantGraph()
1112 result_literal = result_literal.Relayout(*output_layout); in ComputeConstantGraph()
1114 *result->mutable_literal() = result_literal.ToProto(); in ComputeConstantGraph()
|
| D | hlo_evaluator.cc | 473 auto result_literal = LiteralUtil::CreateFromDimensions( in HandleConcatenate() local
480 TF_RETURN_IF_ERROR(result_literal.CopySliceFrom( in HandleConcatenate()
487 evaluated_[concatenate] = std::move(result_literal); in HandleConcatenate()
|
| D | hlo_evaluator_test.cc | 1760 Literal result_literal = in TEST_P() local
1762 EXPECT_TRUE(LiteralTestUtil::Equal(result_literal, result)); in TEST_P()
|
| D | hlo_evaluator_typed_visitor.h | 2936 auto result_literal, in ElementWiseUnaryOp()
2940 return std::move(result_literal); in ElementWiseUnaryOp()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_execute_test.cc | 211 Literal result_literal = ShapedBufferToLiteral(result); in XLA_TEST_F() local
213 LiteralSlice(result_literal, {0})); in XLA_TEST_F()
215 LiteralSlice(result_literal, {1})); in XLA_TEST_F()
217 LiteralSlice(result_literal, {2})); in XLA_TEST_F()
239 Literal result_literal = ShapedBufferToLiteral(result); in XLA_TEST_F() local
241 LiteralSlice(result_literal, {1})); in XLA_TEST_F()
243 LiteralSlice(result_literal, {0, 0})); in XLA_TEST_F()
245 LiteralSlice(result_literal, {0, 1})); in XLA_TEST_F()
247 LiteralSlice(result_literal, {0, 2})); in XLA_TEST_F()
271 Literal result_literal = ShapedBufferToLiteral(result); in XLA_TEST_F() local
[all …]
|
| D | exhaustive_op_test.cc | 273 TF_ASSERT_OK_AND_ASSIGN(Literal result_literal, in RunImpl()
275 ExpectNear<T>(input_literal, result_literal, evaluate_op); in RunImpl()
310 TF_ASSIGN_OR_RETURN(Literal result_literal, in BuildAndRunComputation()
312 return std::move(result_literal); in BuildAndRunComputation()
332 void ExpectNear(const Literal& input_literal, const Literal& result_literal, in ExpectNear() argument
343 absl::Span<const T> result_arr = result_literal.data<T>(); in ExpectNear()
|
| 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 | gather_operation_test.cc | 674 TF_ASSERT_OK_AND_ASSIGN(Literal result_literal, in XLA_TEST_F()
676 LiteralTestUtil::ExpectR2Equal<int32>({{1, 2, 3}, {7, 8, 9}}, result_literal); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable.cc | 97 Literal result_literal; in ExecuteOnStream() local
101 TF_ASSIGN_OR_RETURN(result_literal, in ExecuteOnStream()
108 result_literal.shape(), run_options->allocator(), in ExecuteOnStream()
111 run_options->stream(), result_literal, result)); in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_infeed_test.cc | 171 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/ |
| D | grpc_client_test.cc | 100 TF_ASSERT_OK_AND_ASSIGN(auto result_literal, client_->ExecuteAndTransfer( in TEST_F()
102 EXPECT_TRUE(LiteralTestUtil::Near(expected_literal, result_literal, in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 553 Literal result_literal = in ConvArray4DGeneralDimensionsDilated() local
556 CHECK_EQ(result_literal.shape().rank(), 4); in ConvArray4DGeneralDimensionsDilated()
558 absl::make_unique<Array4D<float>>(result_literal.shape().dimensions(0), in ConvArray4DGeneralDimensionsDilated()
559 result_literal.shape().dimensions(1), in ConvArray4DGeneralDimensionsDilated()
560 result_literal.shape().dimensions(2), in ConvArray4DGeneralDimensionsDilated()
561 result_literal.shape().dimensions(3)); in ConvArray4DGeneralDimensionsDilated()
564 *value = result_literal.Get<float>(indices); in ConvArray4DGeneralDimensionsDilated()
|
| D | literal.cc | 716 Literal result_literal(result_shape); in SliceInternal() local
718 result_literal.EachCell<NativeT>( in SliceInternal()
724 result_literal.Set<NativeT>(indices, value); in SliceInternal()
726 return result_literal; in SliceInternal()
1238 Literal result_literal(ShapeUtil::ChangeElementType( in ConvertBetweenNativeTypesWithConverter() local
1242 auto dest_data = result_literal.template data<NativeDestT>(); in ConvertBetweenNativeTypesWithConverter()
1248 return result_literal; in ConvertBetweenNativeTypesWithConverter()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 304 TF_ASSIGN_OR_RETURN(Literal result_literal, in ReplayComputation()
306 return result_literal; in ReplayComputation()
|