| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_test.cc | 215 auto tuple = LiteralUtil::MakeTuple({&scalar, &matrix}); in TEST_F()
413 EXPECT_EQ(LiteralUtil::MakeTuple({&token0}), in TEST_F()
414 LiteralUtil::MakeTuple({&token0})); in TEST_F()
415 EXPECT_EQ(LiteralUtil::MakeTuple({&token0, &scalar}), in TEST_F()
416 LiteralUtil::MakeTuple({&token1, &scalar})); in TEST_F()
417 EXPECT_NE(LiteralUtil::MakeTuple({&token0, &scalar}), in TEST_F()
418 LiteralUtil::MakeTuple({&scalar, &token1})); in TEST_F()
442 auto tuple1 = LiteralUtil::MakeTuple({&scalar, &matrix}); in TEST_F()
447 auto tuple2 = LiteralUtil::MakeTuple({&scalar_clone, &matrix}); in TEST_F()
451 auto reversed_tuple = LiteralUtil::MakeTuple({&matrix, &scalar}); in TEST_F()
[all …]
|
| D | literal_util.cc | 416 /* static */ Literal LiteralUtil::MakeTuple( in MakeTuple() function in xla::LiteralUtil
|
| D | literal_util.h | 212 static Literal MakeTuple(absl::Span<const Literal* const> elements);
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | all_reduce_test.cc | 71 EXPECT_EQ(LiteralUtil::MakeTuple({&literal0, &literal1}), in XLA_TEST_F()
97 EXPECT_EQ(LiteralUtil::MakeTuple({&literal0, &literal1}), in XLA_TEST_F()
|
| D | while_test.cc | 351 auto expected = LiteralUtil::MakeTuple({&expected_data}); in XLA_TEST_F()
404 auto expected = LiteralUtil::MakeTuple( in XLA_TEST_F()
512 auto expected = LiteralUtil::MakeTuple({&expected_counter, &expected_data}); in XLA_TEST_F()
559 LiteralUtil::MakeTuple({&expected_counter, &expected_predicate}); in XLA_TEST_F()
603 auto expected = LiteralUtil::MakeTuple({&expected_counter, &expected_data}); in XLA_TEST_F()
884 auto expected = LiteralUtil::MakeTuple({&expected_counter, &expected_data}); in XLA_TEST_F()
974 LiteralUtil::MakeTuple({&expected_element, &expected_element}); in XLA_TEST_F()
1071 auto expected = LiteralUtil::MakeTuple({&add1, &add2}); in XLA_TEST_F()
|
| D | tuple_test.cc | 114 auto expected = LiteralUtil::MakeTuple({}); in XLA_TEST_F()
423 LiteralUtil::MakeTuple({&expected_v1, &expected_s}); in XLA_TEST_F()
425 auto expected = LiteralUtil::MakeTuple({&expected_inner_tuple, &expected_v2}); in XLA_TEST_F()
|
| D | round_trip_transfer_test.cc | 110 RoundTripTest(LiteralUtil::MakeTuple({})); in TEST_F()
|
| D | call_test.cc | 143 auto tuple = LiteralUtil::MakeTuple({&elem}); in XLA_TEST_F()
|
| D | custom_call_test.cc | 205 Literal expected = LiteralUtil::MakeTuple({&arg1, &arg0}); in XLA_TEST_F()
|
| D | params_test.cc | 328 ComputeAndCompareTuple(&builder, LiteralUtil::MakeTuple(ptrs), param_data); in XLA_TEST_F()
428 ComputeAndCompareTuple(&builder, LiteralUtil::MakeTuple(ptrs), param_data); in XLA_TEST_F()
|
| D | token_hlo_test.cc | 53 LiteralTestUtil::Equal(result, LiteralUtil::MakeTuple({&token_literal}))); in XLA_TEST_F()
|
| D | transfer_manager_test.cc | 193 Literal literal = LiteralUtil::MakeTuple({}); in XLA_TEST_F()
|
| D | local_client_execute_test.cc | 804 test_to_device_and_back(LiteralUtil::MakeTuple({})); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 69 auto tuple = xla::LiteralUtil::MakeTuple({&array, &matrix}); in TwoElementTuple()
81 auto tuple = xla::LiteralUtil::MakeTuple({&array, &matrix}); in NestedTuple()
83 auto nested = xla::LiteralUtil::MakeTuple({&tuple, &scalar}); in NestedTuple()
91 auto tuple = xla::LiteralUtil::MakeTuple({&array, &matrix}); in MakeTuple0()
92 auto nested0 = xla::LiteralUtil::MakeTuple({&scalar, &tuple}); in MakeTuple0()
93 auto nested1 = xla::LiteralUtil::MakeTuple({&scalar, &nested0}); in MakeTuple0()
277 xla::Literal literal = xla::LiteralUtil::MakeTuple({&literal0, &literal1}); in TEST()
306 xla::Literal literal = xla::LiteralUtil::MakeTuple({&literal0}); in TEST()
313 ops::XRTAllocateFromTensor::Layouts(layout).MakeTuple(true); in TEST()
617 TEST(RawApiTest, MakeTuple) { in TEST() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | infeed_test.cc | 107 TestInfeedRoundTrip(LiteralUtil::MakeTuple({})); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler_test.cc | 228 xla::Literal expected_literal = xla::LiteralUtil::MakeTuple({&expected0}); in TEST_F()
513 xla::Literal expected_literal = xla::LiteralUtil::MakeTuple({&expected0}); in TEST_F()
547 xla::LiteralUtil::MakeTuple({&expected0, &expected1}); in TEST_F()
773 xla::Literal input = xla::LiteralUtil::MakeTuple({&input_base, &input_grad2}); in TEST_F()
787 xla::LiteralUtil::MakeTuple({&output_base, &output_grad1, &output_grad2}); in TEST_F()
789 xla::LiteralUtil::MakeTuple({&output_read, &output_resource}); in TEST_F()
976 xla::LiteralUtil::MakeTuple({&expected0, &expected1}); in RunAndCheckVariablesComputation()
1129 xla::Literal expected_literal = xla::LiteralUtil::MakeTuple({}); in TEST_F()
1256 xla::LiteralUtil::MakeTuple({&expected0, &expected1}); in TEST_F()
1324 xla::LiteralUtil::MakeTuple({&expected0, &expected1}); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.h | 128 static Status MakeTuple(xla::Backend* backend, int device_ordinal,
|
| D | xrt_state.cc | 380 /*static*/ Status XRTTupleAllocation::MakeTuple( in MakeTuple() function in tensorflow::XRTTupleAllocation
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_liveness_test.cc | 445 auto inner_tuple0 = LiteralUtil::MakeTuple({&elements0[0], &elements0[1]}); in TEST_F()
447 auto inner_tuple1 = LiteralUtil::MakeTuple({&element1}); in TEST_F()
449 LiteralUtil::MakeTuple({&inner_tuple0, &inner_tuple1}))); in TEST_F()
|
| D | hlo_evaluator.cc | 775 evaluated_[tuple] = LiteralUtil::MakeTuple(operand_literals); in HandleTuple()
1552 Literal result_tuple = LiteralUtil::MakeTuple(literal_ptrs); in HandleSort()
|
| D | tuple_points_to_analysis_test.cc | 578 LiteralUtil::MakeTuple({&elements[0], &elements[1]}))); in TEST_F()
|
| D | buffer_assignment_test.cc | 1354 LiteralUtil::MakeTuple({&elements[0], &elements[1]}))); in TEST_F()
|
| D | hlo_evaluator_test.cc | 1982 LiteralUtil::MakeTuple({&result_inner_literal, &result_inner_literal}); in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_infeed_test.cc | 106 TestInfeedRoundTrip(LiteralUtil::MakeTuple({})); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 429 OP_REQUIRES_OK(ctx, XRTTupleAllocation::MakeTuple( in Compute()
|