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 | 254 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()
|
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 | 324 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()
|
D | reduce_window_test.cc | 1244 input = *MakeLinspaceArray2D(0, 100, param.base_bounds[0], in DoIt()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_runtime_test.cc | 156 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/ |
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 | 340 auto value = MakeLinspaceArray2D(from, to, rows, cols); in CreateR2F32Linspace()
|