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.cc48 std::unique_ptr<GlobalData> param0_data = CreateR1Parameter<float>( in XLA_TEST_F()
53 std::unique_ptr<GlobalData> param1_data = CreateR1Parameter<float>( in XLA_TEST_F()
70 std::unique_ptr<GlobalData> param0_data = CreateR1Parameter<float>( in XLA_TEST_F()
75 std::unique_ptr<GlobalData> param1_data = CreateR1Parameter<float>( in XLA_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.cc350 CreateR1Parameter<float>(std::vector<float>(260, 1.0f), in XLA_TEST_P()
354 CreateR1Parameter<float>(std::vector<float>(260, 1.0f), in XLA_TEST_P()
381 CreateR1Parameter<float>(std::vector<float>(1, 1.0f), in XLA_TEST_P()
385 CreateR1Parameter<float>(std::vector<float>(1, 0.0f), in XLA_TEST_P()
Dclient_library_test_base.h350 std::unique_ptr<GlobalData> CreateR1Parameter(
619 std::unique_ptr<GlobalData> ClientLibraryTestBase::CreateR1Parameter( in CreateR1Parameter() function
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()
Darray_elementwise_ops_test.cc532 CreateR1Parameter<T>(dividends, 0, "dividend", &builder, &dividend); in TestDivRem()
534 CreateR1Parameter<T>(divisors, 1, "divisor", &builder, &divisor); in TestDivRem()
546 CreateR1Parameter<T>(dividends, 0, "dividend", &builder, &dividend); in TestDivRem()
557 CreateR1Parameter<T>(dividends, 0, "dividend", &builder, &dividend); in TestDivRem()
559 CreateR1Parameter<T>(divisors, 1, "divisor", &builder, &divisor); in TestDivRem()
571 CreateR1Parameter<T>(dividends, 0, "dividend", &builder, &dividend); in TestDivRem()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dslicing_test.cc163 CreateR1Parameter<int>({0, 2}, 1, "index", &builder, &index); in XLA_TEST_F()
182 CreateR1Parameter<int>({0, 2}, 1, "index", &builder, &index); in XLA_TEST_F()
197 auto index_data = CreateR1Parameter<int>({}, 1, "index", &builder, &index); in XLA_TEST_F()
228 CreateR1Parameter<int>({0, 0, 0}, 1, "index", &builder, &index); in XLA_TEST_F()
Dsorting_test.cc74 auto a_data = CreateR1Parameter<int>({1, 1, 2, 2, 1}, 0, "a", &builder, &a); in XLA_TEST_F()
146 auto a_data = CreateR1Parameter<int>({1, 1, 2, 2, 1}, 0, "a", &builder, &a); in XLA_TEST_F()