Home
last modified time | relevance | path

Searched refs:CreateR0Parameter (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconditional_test.cc182 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 …]
Ddeep_graph_test.cc29 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()
Ddynamic_ops_test.cc139 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()
Dscalar_computations_test.cc312 CreateR0Parameter<int32>(p.dividend, 0, "dividend", &builder, &dividend); 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, &dividend); in XLA_TEST_P()
329 CreateR0Parameter<int32>(p.divisor, 1, "divisor", &builder, &divisor); in XLA_TEST_P()
Dtuple_test.cc211 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()
Dclient_library_test_base.h335 std::unique_ptr<GlobalData> CreateR0Parameter(NativeT value,
605 std::unique_ptr<GlobalData> ClientLibraryTestBase::CreateR0Parameter( in CreateR0Parameter() function
Dbroadcast_simple_test.cc125 CreateR0Parameter<float>(2.25f, /*parameter_number=*/0, /*name=*/"src", in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dslicing_test.cc64 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()
Dmatrix_test.cc188 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()