Searched refs:MakeLinspaceArray2D (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | binop_scaling_test.cc | 32 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()
|
| D | reshape_test.cc | 241 auto simple = MakeLinspaceArray2D(1.0f, 3.0f, 1, 3); in XLA_TEST_P()
258 auto a4x3 = MakeLinspaceArray2D(1.0f, 12.0f, 4, 3); in XLA_TEST_P()
288 auto a4x3 = MakeLinspaceArray2D(1.0f, 12.0f, 4, 3); in XLA_TEST_P()
334 auto a4x3 = MakeLinspaceArray2D(1.0f, 12.0f, 4, 3); in XLA_TEST_P()
342 auto expected = MakeLinspaceArray2D(1.0f, 12.0f, 2, 6); in XLA_TEST_P()
365 auto a4x3 = MakeLinspaceArray2D(1.0f, 12.0f, 4, 3); in XLA_TEST_P()
493 auto filler2x3 = MakeLinspaceArray2D(1.0f, 6.0f, 2, 3); in XLA_TEST_P()
|
| D | transpose_test.cc | 135 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()
|
| D | constants_test.cc | 106 MakeLinspaceArray2D(100.0, 200.0, 2, 2); in TEST_F()
|
| D | matrix_ops_simple_test.cc | 136 MakeLinspaceArray2D<T>(from, to, rows, cols); in TestImpl()
|
| D | dot_operation_test.cc | 289 MakeLinspaceArray2D<NativeT>(0.0, 1.0, param.m, param.k); in TestImpl()
297 MakeLinspaceArray2D<NativeT>(0.0, 1.0, param.k, param.n); in TestImpl()
310 addend_data = MakeLinspaceArray2D<NativeT>(0.0, 1.0, param.m, param.n); in TestImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime_test.cc | 158 auto a = MakeLinspaceArray2D(0.0, 1.0, shape.m, shape.k); in TEST_P()
159 auto b = MakeLinspaceArray2D(-2.0, 2.0, shape.k, shape.n); in TEST_P()
227 auto a = MakeLinspaceArray2D(0.0, 1.0, shape.m, shape.k); in TEST_P()
228 auto b = MakeLinspaceArray2D(-2.0, 2.0, shape.k, shape.n); in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | array2d_test.cc | 135 auto arr = MakeLinspaceArray2D(1.0, 3.5, 3, 2); in TEST()
149 auto arr = MakeLinspaceArray2D(1.0, 3.5, 3, 2); in TEST()
|
| D | array2d.h | 101 std::unique_ptr<Array2D<NativeT>> MakeLinspaceArray2D(double from, double to, in MakeLinspaceArray2D() function
|
| D | literal_util.cc | 288 auto value = MakeLinspaceArray2D(from, to, rows, cols); in CreateR2F32Linspace()
|