Searched defs:matrix_shape (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | matrix_solve_op_test.py | 156 def _GenerateTestData(self, matrix_shape, num_rhs): argument
|
| D | matrix_solve_ls_op_test.py | 45 def _GenerateTestData(matrix_shape, num_rhs): argument
|
| D | tridiagonal_solve_op_test.py | 381 def test_with_matrix_shapes(matrix_shape): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 104 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
303 Shape matrix_shape = ShapeUtil::MakeShape(F32, {8, 8}); in TEST_F() local
834 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
843 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
852 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
861 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
870 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | xla_hlo_profile_test.cc | 270 Shape matrix_shape = ShapeUtil::MakeShape(F32, {size, size}); in XLA_TEST_F() local
|
| D | tuple_test.cc | 171 auto matrix_shape = builder.GetShape(matrix_element).ConsumeValueOrDie(); in XLA_TEST_F() local
|
| D | while_test.cc | 1190 auto matrix_shape = ShapeUtil::MakeShape(F32, {2, 2}); in XLA_TEST_F() local
|