Searched refs:CreateParameterAndTransferLiteral (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reshape_test.cc | 61 auto input, CreateParameterAndTransferLiteral( in XLA_TEST_P()
75 auto input, CreateParameterAndTransferLiteral( in XLA_TEST_P()
89 auto input, CreateParameterAndTransferLiteral( in XLA_TEST_P()
106 auto input, CreateParameterAndTransferLiteral( in XLA_TEST_P()
123 auto input, CreateParameterAndTransferLiteral(0, param0_literal, "param0", in XLA_TEST_P()
139 auto input, CreateParameterAndTransferLiteral(0, input_literal, "input", in XLA_TEST_P()
154 auto input, CreateParameterAndTransferLiteral(0, param0_literal, "param0", in XLA_TEST_P()
168 auto input, CreateParameterAndTransferLiteral(0, input_literal, "input", in XLA_TEST_P()
182 auto input, CreateParameterAndTransferLiteral(0, input_literal, "input", in XLA_TEST_P()
196 auto input, CreateParameterAndTransferLiteral(0, input_literal, "input", in XLA_TEST_P()
[all …]
|
| D | reduce_window_test.cc | 369 auto input_data, CreateParameterAndTransferLiteral( in XLA_TEST_P()
428 auto input_data, CreateParameterAndTransferLiteral( in XLA_TEST_P()
450 auto input_data, CreateParameterAndTransferLiteral( in XLA_TEST_P()
472 auto input_data, CreateParameterAndTransferLiteral( in XLA_TEST_P()
639 CreateParameterAndTransferLiteral( in DoIt()
1251 TF_ASSERT_OK(CreateParameterAndTransferLiteral(0, input_literal, "p0", &b, in DoIt()
1454 auto input_arg, CreateParameterAndTransferLiteral(0, input_literal, "p0", in XLA_TEST_P()
|
| D | client_library_test_base.cc | 609 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
|
| D | client_library_test_base.h | 279 StatusOr<std::unique_ptr<GlobalData>> CreateParameterAndTransferLiteral(
286 StatusOr<std::unique_ptr<GlobalData>> CreateParameterAndTransferLiteral(
|
| D | dot_operation_test.cc | 77 CreateParameterAndTransferLiteral( in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing_test.cc | 211 CreateParameterAndTransferLiteral(0, l, "input", &builder, &input)); in XLA_TEST_F()
214 CreateParameterAndTransferLiteral(1, i, "index", &builder, &index)); in XLA_TEST_F()
226 CreateParameterAndTransferLiteral(0, l, "input", &builder, &input)); in XLA_TEST_F()
|