Home
last modified time | relevance | path

Searched refs:MakeLinspaceArray2D (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dbinop_scaling_test.cc32 auto alhs = MakeLinspaceArray2D(0.0, 1.0, 32, 4); in TEST_F()
33 auto arhs = MakeLinspaceArray2D(0.0, 1.0, 1, 4); in TEST_F()
48 auto alhs = MakeLinspaceArray2D(0.0, 1.0, 129, 129); in TEST_F()
49 auto arhs = MakeLinspaceArray2D(0.0, 1.0, 1, 129); in TEST_F()
64 auto alhs = MakeLinspaceArray2D(0.0, 1.0, 9, 5); in TEST_F()
65 auto arhs = MakeLinspaceArray2D(0.0, 1.0, 9, 1); in TEST_F()
80 auto alhs = MakeLinspaceArray2D(0.0, 1.0, 129, 257); in TEST_F()
81 auto arhs = MakeLinspaceArray2D(0.0, 1.0, 129, 1); in TEST_F()
Dtranspose_test.cc135 auto aoperand = MakeLinspaceArray2D(0.0, 1.0, 1, 1); in TEST_F()
147 auto aoperand = MakeLinspaceArray2D(0.0, 4.0, 2, 2); in TEST_F()
Dreshape_test.cc254 auto simple = MakeLinspaceArray2D(1.0f, 3.0f, 1, 3); in XLA_TEST_P()
272 auto a4x3 = MakeLinspaceArray2D(1.0f, 12.0f, 4, 3); in XLA_TEST_P()
304 auto a4x3 = MakeLinspaceArray2D(1.0f, 12.0f, 4, 3); in XLA_TEST_P()
353 auto a4x3 = MakeLinspaceArray2D(1.0f, 12.0f, 4, 3); in XLA_TEST_P()
362 auto expected = MakeLinspaceArray2D(1.0f, 12.0f, 2, 6); in XLA_TEST_P()
386 auto a4x3 = MakeLinspaceArray2D(1.0f, 12.0f, 4, 3); in XLA_TEST_P()
520 auto filler2x3 = MakeLinspaceArray2D(1.0f, 6.0f, 2, 3); in XLA_TEST_P()
Dconstants_test.cc106 MakeLinspaceArray2D(100.0, 200.0, 2, 2); in TEST_F()
Dmatrix_ops_simple_test.cc136 MakeLinspaceArray2D<T>(from, to, rows, cols); in TestImpl()
Ddot_operation_test.cc324 MakeLinspaceArray2D<NativeT>(0.0, 1.0, param.m, param.k); in TestImpl()
332 MakeLinspaceArray2D<NativeT>(0.0, 1.0, param.k, param.n); in TestImpl()
345 addend_data = MakeLinspaceArray2D<NativeT>(0.0, 1.0, param.m, param.n); in TestImpl()
Dreduce_window_test.cc1244 input = *MakeLinspaceArray2D(0, 100, param.base_bounds[0], in DoIt()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime_test.cc156 auto a = MakeLinspaceArray2D(0.0, 1.0, shape.m, shape.k); in TEST_P()
157 auto b = MakeLinspaceArray2D(-2.0, 2.0, shape.k, shape.n); in TEST_P()
225 auto a = MakeLinspaceArray2D(0.0, 1.0, shape.m, shape.k); in TEST_P()
226 auto b = MakeLinspaceArray2D(-2.0, 2.0, shape.k, shape.n); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/
Darray2d_test.cc135 auto arr = MakeLinspaceArray2D(1.0, 3.5, 3, 2); in TEST()
149 auto arr = MakeLinspaceArray2D(1.0, 3.5, 3, 2); in TEST()
Darray2d.h101 std::unique_ptr<Array2D<NativeT>> MakeLinspaceArray2D(double from, double to, in MakeLinspaceArray2D() function
Dliteral_util.cc340 auto value = MakeLinspaceArray2D(from, to, rows, cols); in CreateR2F32Linspace()