Home
last modified time | relevance | path

Searched refs:MakeTuple (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dall_reduce_test.cc72 EXPECT_EQ(LiteralUtil::MakeTuple({&literal0, &literal1}), in XLA_TEST_F()
99 EXPECT_EQ(LiteralUtil::MakeTuple({&literal0, &literal1}), in XLA_TEST_F()
140 EXPECT_EQ(LiteralUtil::MakeTuple({&literal0}), in XLA_TEST_F()
182 EXPECT_EQ(LiteralUtil::MakeTuple({&literal0}), in XLA_TEST_F()
Dwhile_test.cc351 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()
Dround_trip_transfer_test.cc110 RoundTripTest(LiteralUtil::MakeTuple({})); in TEST_F()
Dtuple_test.cc114 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()
Dcall_test.cc143 auto tuple = LiteralUtil::MakeTuple({&elem}); in XLA_TEST_F()
Dcustom_call_test.cc205 Literal expected = LiteralUtil::MakeTuple({&arg1, &arg0}); in XLA_TEST_F()
Dtoken_hlo_test.cc52 LiteralTestUtil::Equal(result, LiteralUtil::MakeTuple({&token_literal}))); in XLA_TEST_F()
Dparams_test.cc328 ComputeAndCompareTuple(&builder, LiteralUtil::MakeTuple(ptrs), param_data); in XLA_TEST_F()
428 ComputeAndCompareTuple(&builder, LiteralUtil::MakeTuple(ptrs), param_data); in XLA_TEST_F()
Dtransfer_manager_test.cc235 Literal literal = LiteralUtil::MakeTuple({}); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_test.cc225 auto tuple = LiteralUtil::MakeTuple({&scalar, &matrix}); in TEST_F()
387 EXPECT_EQ(LiteralUtil::MakeTuple({&token0}), in TEST_F()
388 LiteralUtil::MakeTuple({&token0})); in TEST_F()
389 EXPECT_EQ(LiteralUtil::MakeTuple({&token0, &scalar}), in TEST_F()
390 LiteralUtil::MakeTuple({&token1, &scalar})); in TEST_F()
391 EXPECT_NE(LiteralUtil::MakeTuple({&token0, &scalar}), in TEST_F()
392 LiteralUtil::MakeTuple({&scalar, &token1})); in TEST_F()
416 auto tuple1 = LiteralUtil::MakeTuple({&scalar, &matrix}); in TEST_F()
421 auto tuple2 = LiteralUtil::MakeTuple({&scalar_clone, &matrix}); in TEST_F()
425 auto reversed_tuple = LiteralUtil::MakeTuple({&matrix, &scalar}); in TEST_F()
[all …]
Dliteral_util.h175 static Literal MakeTuple(absl::Span<const Literal* const> elements);
Dliteral_util.cc468 /* static */ Literal LiteralUtil::MakeTuple( in MakeTuple() function in xla::LiteralUtil
/external/llvm-project/clang-tools-extra/clang-tidy/modernize/
DUseEmplaceCheck.cpp93 auto MakeTuple = ignoringImplicit( in registerMatchers() local
104 has(materializeTemporaryExpr(MakeTuple)), in registerMatchers()
109 anyOf(has(MakeTuple), has(MakeTupleCtor), in registerMatchers()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dinfeed_test.cc107 TestInfeedRoundTrip(LiteralUtil::MakeTuple({})); in TEST_F()
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc165 auto tuple = xla::LiteralUtil::MakeTuple({&array, &matrix}); in TwoElementTuple()
173 auto tuple = xla::LiteralUtil::MakeTuple({&array, &matrix}); in BasedTwoElementTuple()
185 auto tuple = xla::LiteralUtil::MakeTuple({&array, &matrix}); in NestedTuple()
187 auto nested = xla::LiteralUtil::MakeTuple({&tuple, &scalar}); in NestedTuple()
195 auto tuple = xla::LiteralUtil::MakeTuple({&array, &matrix}); in MakeTuple0()
196 auto nested0 = xla::LiteralUtil::MakeTuple({&scalar, &tuple}); in MakeTuple0()
197 auto nested1 = xla::LiteralUtil::MakeTuple({&scalar, &nested0}); in MakeTuple0()
482 xla::Literal literal = xla::LiteralUtil::MakeTuple({&literal0, &literal1}); in TEST()
511 xla::Literal literal = xla::LiteralUtil::MakeTuple({&literal0}); in TEST()
518 ops::XRTAllocateFromTensor::Layouts(layout).MakeTuple(true); in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc225 return MakeTuple(std::move(child_buffer_ptrs), std::move(client), in FromLiterals()
230 StatusOr<std::unique_ptr<PyTpuBuffer>> PyTpuBuffer::MakeTuple( in MakeTuple() function in xla::PyTpuBuffer
466 return PyTpuBuffer::MakeTuple(child_buffer_ptrs, std::move(client), in AllocateBuffer()
639 PyTpuBuffer::MakeTuple(argument_handles, client_, in Execute()
693 PyTpuBuffer::MakeTuple(argument_handles[i], client_, in ExecuteOnLocalDevices()
Dtpu_client.h193 static StatusOr<std::unique_ptr<PyTpuBuffer>> MakeTuple(
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Dcustom_call.md128 Shape p0_shape = ShapeUtil::MakeTuple({
130 ShapeUtil::MakeTuple({
138 Shape out_shape = ShapeUtil::MakeTuple({
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc235 xla::Literal expected_literal = xla::LiteralUtil::MakeTuple({&expected0}); in TEST_F()
623 xla::LiteralUtil::MakeTuple({&expected0, &expected1}); in TEST_F()
841 xla::Literal input = xla::LiteralUtil::MakeTuple({&input_base, &input_grad2}); in TEST_F()
855 xla::LiteralUtil::MakeTuple({&output_base, &output_grad1, &output_grad2}); in TEST_F()
857 xla::LiteralUtil::MakeTuple({&output_read, &output_resource}); in TEST_F()
1041 xla::LiteralUtil::MakeTuple({&expected0, &expected1}); in RunAndCheckVariablesComputation()
1197 xla::Literal expected_literal = xla::LiteralUtil::MakeTuple({}); in TEST_F()
1324 xla::LiteralUtil::MakeTuple({&expected0, &expected1}); in TEST_F()
1393 xla::LiteralUtil::MakeTuple({&expected0, &expected1}); in TEST_F()
1749 xla::LiteralUtil::MakeTuple({&expected0, &expected1, &expected2}); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.h139 static Status MakeTuple(XRTMemoryManager* memory_manager,
Dxrt_state.cc454 /*static*/ Status XRTTupleAllocation::MakeTuple( in MakeTuple() function in tensorflow::XRTTupleAllocation
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_infeed_test.cc106 TestInfeedRoundTrip(LiteralUtil::MakeTuple({})); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc812 evaluated_[tuple] = LiteralUtil::MakeTuple(operand_literals); in HandleTuple()
1893 evaluated_[copy_start] = LiteralUtil::MakeTuple( in HandleCopyStart()
2272 Literal result_tuple = LiteralUtil::MakeTuple(literal_ptrs); in HandleSort()
Dtuple_points_to_analysis_test.cc606 LiteralUtil::MakeTuple({&elements[0], &elements[1]}))); in TEST_F()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h465 OP_REQUIRES_OK(ctx, XRTTupleAllocation::MakeTuple( in Compute()

12