Home
last modified time | relevance | path

Searched refs:CreateParameterAndTransferLiteral (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dreshape_test.cc60 auto input = CreateParameterAndTransferLiteral(0, input_literal, "parameter", in XLA_TEST_P()
73 auto input = CreateParameterAndTransferLiteral(0, input_literal, "parameter", in XLA_TEST_P()
86 auto input = CreateParameterAndTransferLiteral(0, input_literal, "parameter", in XLA_TEST_P()
102 auto input = CreateParameterAndTransferLiteral(0, input_literal, "parameter", in XLA_TEST_P()
118 auto input = CreateParameterAndTransferLiteral(0, param0_literal, "param0", in XLA_TEST_P()
133 auto input = CreateParameterAndTransferLiteral(0, input_literal, "input", in XLA_TEST_P()
147 auto input = CreateParameterAndTransferLiteral(0, param0_literal, "param0", in XLA_TEST_P()
160 auto input = CreateParameterAndTransferLiteral(0, input_literal, "input", in XLA_TEST_P()
173 auto input = CreateParameterAndTransferLiteral(0, input_literal, "input", in XLA_TEST_P()
186 auto input = CreateParameterAndTransferLiteral(0, input_literal, "input", in XLA_TEST_P()
[all …]
Dreduce_window_test.cc345 auto input_data = CreateParameterAndTransferLiteral( in TEST_P()
403 auto input_data = CreateParameterAndTransferLiteral( in XLA_TEST_P()
424 auto input_data = CreateParameterAndTransferLiteral( in XLA_TEST_P()
445 auto input_data = CreateParameterAndTransferLiteral( in XLA_TEST_P()
623 auto input_arg = CreateParameterAndTransferLiteral(0, input_literal, "p0", in DoIt()
1167 auto input_arg = CreateParameterAndTransferLiteral(0, input_literal, "p0", in DoIt()
1361 CreateParameterAndTransferLiteral(0, input_literal, "p0", &b, &parameter); in TEST_P()
Dclient_library_test_base.cc609 ClientLibraryTestBase::CreateParameterAndTransferLiteral(int64 parameter_number, in CreateParameterAndTransferLiteral() function in xla::ClientLibraryTestBase
614 return CreateParameterAndTransferLiteral(parameter_number, literal, name, in CreateParameterAndTransferLiteral()
641 ClientLibraryTestBase::CreateParameterAndTransferLiteral( in CreateParameterAndTransferLiteral() function in xla::ClientLibraryTestBase
Dclient_library_test_base.h271 std::unique_ptr<GlobalData> CreateParameterAndTransferLiteral(
278 std::unique_ptr<GlobalData> CreateParameterAndTransferLiteral(
Ddot_operation_test.cc70 auto param_data = CreateParameterAndTransferLiteral( in XLA_TEST_F()