Home
last modified time | relevance | path

Searched refs:CreateR1Parameter (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dfmax_fmin_test.cc47 std::unique_ptr<GlobalData> param0_data = CreateR1Parameter<float>( in TEST_F()
52 std::unique_ptr<GlobalData> param1_data = CreateR1Parameter<float>( in TEST_F()
69 std::unique_ptr<GlobalData> param0_data = CreateR1Parameter<float>( in TEST_F()
74 std::unique_ptr<GlobalData> param1_data = CreateR1Parameter<float>( in TEST_F()
Dhalf_test.cc55 auto x_data = CreateR1Parameter<half>(x, /*parameter_number=*/0, "x", in XLA_TEST_P()
109 auto x_data = CreateR1Parameter<half>(x, /*parameter_number=*/0, "x", in XLA_TEST_P()
145 auto x_data = CreateR1Parameter<half>(x, /*parameter_number=*/0, "x", in XLA_TEST_P()
149 auto y_data = CreateR1Parameter<half>(y, /*parameter_number=*/1, "y", in XLA_TEST_P()
199 auto x_data = CreateR1Parameter<half>(x, /*parameter_number=*/0, "x", in XLA_TEST_P()
203 auto y_data = CreateR1Parameter<half>(y, /*parameter_number=*/1, "y", in XLA_TEST_P()
Dselect_test.cc142 std::unique_ptr<GlobalData> param0_data = CreateR1Parameter<float>( in TEST_F()
145 std::unique_ptr<GlobalData> param1_data = CreateR1Parameter<float>( in TEST_F()
184 CreateR1Parameter<float>(v1vec, /*parameter_number=*/0, /*name=*/"v1", in TEST_F()
187 CreateR1Parameter<float>(v2vec, /*parameter_number=*/1, /*name=*/"v2", in TEST_F()
Dvector_ops_simple_test.cc183 std::unique_ptr<GlobalData> param0_data = CreateR1Parameter<float>( in XLA_TEST_F()
186 std::unique_ptr<GlobalData> param1_data = CreateR1Parameter<float>( in XLA_TEST_F()
224 CreateR1Parameter<float>(v1vec, /*parameter_number=*/0, /*name=*/"v1", in XLA_TEST_F()
227 CreateR1Parameter<float>(v2vec, /*parameter_number=*/1, /*name=*/"v2", in XLA_TEST_F()
Dbatch_normalization_test.cc289 CreateR1Parameter<float>(std::vector<float>(260, 1.0f), in XLA_TEST_P()
293 CreateR1Parameter<float>(std::vector<float>(260, 1.0f), in XLA_TEST_P()
320 CreateR1Parameter<float>(std::vector<float>(1, 1.0f), in XLA_TEST_P()
324 CreateR1Parameter<float>(std::vector<float>(1, 0.0f), in XLA_TEST_P()
Dclient_library_test_base.h342 std::unique_ptr<GlobalData> CreateR1Parameter(
563 std::unique_ptr<GlobalData> ClientLibraryTestBase::CreateR1Parameter( in CreateR1Parameter() function
Darray_elementwise_ops_test.cc477 CreateR1Parameter<T>(dividends, 0, "dividend", &builder, &dividend); in TestDivRem()
479 CreateR1Parameter<T>(divisors, 1, "divisor", &builder, &divisor); in TestDivRem()
491 CreateR1Parameter<T>(dividends, 0, "dividend", &builder, &dividend); in TestDivRem()
502 CreateR1Parameter<T>(dividends, 0, "dividend", &builder, &dividend); in TestDivRem()
504 CreateR1Parameter<T>(divisors, 1, "divisor", &builder, &divisor); in TestDivRem()
516 CreateR1Parameter<T>(dividends, 0, "dividend", &builder, &dividend); in TestDivRem()
Dconditional_test.cc617 auto operand1_param = CreateR1Parameter<float>({24.3f, 56.7f}, 1, "operand1", in XLA_TEST_F()
619 auto operand2_param = CreateR1Parameter<float>({10.2f, 11.6f}, 2, "operand2", in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dslicing_test.cc129 CreateR1Parameter<int>({0, 2}, 1, "index", &builder, &index); in XLA_TEST_F()
148 CreateR1Parameter<int>({0, 2}, 1, "index", &builder, &index); in XLA_TEST_F()
Dsorting_test.cc87 auto a_data = CreateR1Parameter<int>({1, 1, 2, 2, 1}, 0, "a", &builder, &a); in XLA_TEST_F()