Searched refs:MakeLinspaceArray2D (Results 1 – 11 of 11) 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 | 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 | reshape_test.cc | 253 auto simple = MakeLinspaceArray2D(1.0f, 3.0f, 1, 3); in XLA_TEST_P()
271 auto a4x3 = MakeLinspaceArray2D(1.0f, 12.0f, 4, 3); in XLA_TEST_P()
303 auto a4x3 = MakeLinspaceArray2D(1.0f, 12.0f, 4, 3); in XLA_TEST_P()
352 auto a4x3 = MakeLinspaceArray2D(1.0f, 12.0f, 4, 3); in XLA_TEST_P()
361 auto expected = MakeLinspaceArray2D(1.0f, 12.0f, 2, 6); in XLA_TEST_P()
385 auto a4x3 = MakeLinspaceArray2D(1.0f, 12.0f, 4, 3); in XLA_TEST_P()
519 auto filler2x3 = MakeLinspaceArray2D(1.0f, 6.0f, 2, 3); in XLA_TEST_P()
|
| D | constants_test.cc | 106 MakeLinspaceArray2D(100.0, 200.0, 2, 2); in TEST_F()
|
| D | matrix_ops_simple_test.cc | 134 MakeLinspaceArray2D<T>(from, to, rows, cols); in TestImpl()
|
| D | dot_operation_test.cc | 323 MakeLinspaceArray2D<NativeT>(0.0, 1.0, param.m, param.k); in TestImpl()
331 MakeLinspaceArray2D<NativeT>(0.0, 1.0, param.k, param.n); in TestImpl()
344 addend_data = MakeLinspaceArray2D<NativeT>(0.0, 1.0, param.m, param.n); in TestImpl()
|
| D | reduce_window_test.cc | 1293 input = *MakeLinspaceArray2D(0, 100, param.base_bounds[0], in DoIt()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime_test.cc | 159 auto a = MakeLinspaceArray2D(0.0, 1.0, shape.m, shape.k); in TEST_P()
160 auto b = MakeLinspaceArray2D(-2.0, 2.0, shape.k, shape.n); in TEST_P()
229 auto a = MakeLinspaceArray2D(0.0, 1.0, shape.m, shape.k); in TEST_P()
230 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 | 98 std::unique_ptr<Array2D<NativeT>> MakeLinspaceArray2D(double from, double to, in MakeLinspaceArray2D() function
|
| D | literal_util.cc | 384 auto value = MakeLinspaceArray2D(from, to, rows, cols); in CreateR2F32Linspace()
|