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 | 139 std::unique_ptr<GlobalData> start_data = CreateR0Parameter<IndexT>( in RunR1() 167 start_data[i] = CreateR0Parameter<IndexT>( in RunR2() 202 start_data[i] = CreateR0Parameter<IndexT>( in RunR3() 422 std::unique_ptr<GlobalData> start_data = CreateR0Parameter<IndexT>( in RunR1() 455 start_data[i] = CreateR0Parameter<IndexT>( in RunR2() 494 start_data[i] = CreateR0Parameter<IndexT>( in RunR3()
|
D | scalar_computations_test.cc | 312 CreateR0Parameter<int32>(p.dividend, 0, "dividend", &builder, ÷nd); in XLA_TEST_P() 314 CreateR0Parameter<int32>(p.divisor, 1, "divisor", &builder, &divisor); in XLA_TEST_P() 327 CreateR0Parameter<int32>(p.dividend, 0, "dividend", &builder, ÷nd); in XLA_TEST_P() 329 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 | 335 std::unique_ptr<GlobalData> CreateR0Parameter(NativeT value, 605 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 | 64 auto x_data = CreateR0Parameter<int>(2, 1, "x", &builder, &x); in XLA_TEST_F() 65 auto y_data = CreateR0Parameter<int>(1, 2, "y", &builder, &y); in XLA_TEST_F() 79 auto index_data = CreateR0Parameter<int>(1, 1, "index", &builder, &index); in XLA_TEST_F() 94 auto x_data = CreateR0Parameter<int>(2, 2, "x", &builder, &x); in XLA_TEST_F() 95 auto y_data = CreateR0Parameter<int>(1, 3, "y", &builder, &y); in XLA_TEST_F()
|
D | matrix_test.cc | 188 auto index_data = CreateR0Parameter<int>(1, 2, "index", &builder, &index); in XLA_TEST_F() 209 auto index_data = CreateR0Parameter<int>(1, 2, "index", &builder, &index); in XLA_TEST_F()
|