Searched refs:CreateR0Parameter (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conditional_test.cc | 182 auto pred_arg = CreateR0Parameter<bool>(true, 0, "pred", &builder, &pred); in XLA_TEST_F()
198 auto branch_index_arg = CreateR0Parameter<int32>(bi, 0, "branch_index_arg", in XLA_TEST_P()
225 auto pred_arg = CreateR0Parameter<bool>(false, 0, "pred", &builder, &pred); in XLA_TEST_F()
241 auto branch_index_arg = CreateR0Parameter<int32>(bi, 0, "branch_index_arg", in XLA_TEST_P()
278 auto pred_arg = CreateR0Parameter<bool>(false, 0, "pred", &builder, &pred); in XLA_TEST_F()
292 auto pred_arg = CreateR0Parameter<bool>(false, 0, "pred", &builder, &pred); in XLA_TEST_F()
305 auto pred_arg = CreateR0Parameter<bool>(false, 0, "pred", &builder, &pred); in XLA_TEST_F()
319 auto pred_arg = CreateR0Parameter<bool>(false, 0, "pred", &builder, &pred); in XLA_TEST_F()
332 auto pred_arg = CreateR0Parameter<bool>(false, 0, "pred", &builder, &pred); in XLA_TEST_F()
354 auto pred_arg = CreateR0Parameter<bool>(false, 0, "pred", &builder, &pred); in XLA_TEST_F()
[all …]
|
| D | deep_graph_test.cc | 29 auto x_data = CreateR0Parameter<int32>(3, 0, "x", &b, &x); in TEST_F()
30 auto y_data = CreateR0Parameter<int32>(1, 1, "y", &b, &y); in TEST_F()
|
| D | dynamic_ops_test.cc | 138 std::unique_ptr<GlobalData> start_data = CreateR0Parameter<IndexT>( in RunR1()
166 start_data[i] = CreateR0Parameter<IndexT>( in RunR2()
201 start_data[i] = CreateR0Parameter<IndexT>( in RunR3()
421 std::unique_ptr<GlobalData> start_data = CreateR0Parameter<IndexT>( in RunR1()
454 start_data[i] = CreateR0Parameter<IndexT>( in RunR2()
493 start_data[i] = CreateR0Parameter<IndexT>( in RunR3()
|
| D | scalar_computations_test.cc | 302 CreateR0Parameter<int32>(p.dividend, 0, "dividend", &builder, ÷nd); in XLA_TEST_P()
304 CreateR0Parameter<int32>(p.divisor, 1, "divisor", &builder, &divisor); in XLA_TEST_P()
317 CreateR0Parameter<int32>(p.dividend, 0, "dividend", &builder, ÷nd); in XLA_TEST_P()
319 CreateR0Parameter<int32>(p.divisor, 1, "divisor", &builder, &divisor); in XLA_TEST_P()
|
| D | tuple_test.cc | 211 CreateR0Parameter<float>(0.0f, /*parameter_number=*/0, /*name=*/"v1", in XLA_TEST_F()
214 CreateR0Parameter<float>(1.0f, /*parameter_number=*/1, /*name=*/"v2", in XLA_TEST_F()
|
| D | client_library_test_base.h | 327 std::unique_ptr<GlobalData> CreateR0Parameter(NativeT value,
549 std::unique_ptr<GlobalData> ClientLibraryTestBase::CreateR0Parameter( in CreateR0Parameter() function
|
| D | broadcast_simple_test.cc | 125 CreateR0Parameter<float>(2.25f, /*parameter_number=*/0, /*name=*/"src", in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing_test.cc | 62 auto x_data = CreateR0Parameter<int>(2, 1, "x", &builder, &x); in XLA_TEST_F()
63 auto y_data = CreateR0Parameter<int>(1, 2, "y", &builder, &y); in XLA_TEST_F()
77 auto index_data = CreateR0Parameter<int>(1, 1, "index", &builder, &index); in XLA_TEST_F()
92 auto x_data = CreateR0Parameter<int>(2, 2, "x", &builder, &x); in XLA_TEST_F()
93 auto y_data = CreateR0Parameter<int>(1, 3, "y", &builder, &y); in XLA_TEST_F()
|
| D | matrix_test.cc | 108 auto index_data = CreateR0Parameter<int>(1, 2, "index", &builder, &index); in XLA_TEST_F()
129 auto index_data = CreateR0Parameter<int>(1, 2, "index", &builder, &index); in XLA_TEST_F()
|