Searched defs:matrix_shape (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | sparse_csr_matrix_ops.cc | 45 const ShapeHandle& matrix_shape, in ValidateSquareMatrixShape()
558 ShapeHandle matrix_shape = sparse_matrix_shape_and_type.shape; in __anona88575120f02() local
580 ShapeHandle matrix_shape = sparse_matrix_shape_and_type.shape; in __anona88575121002() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | shape_inference_test.cc | 108 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
317 Shape matrix_shape = ShapeUtil::MakeShape(F32, {8, 8}); in TEST_F() local
1306 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
1315 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}, {true, true}); in TEST_F() local
1325 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
1334 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
1343 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
1352 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | matrix_solve_op.cc | 32 const TensorShape matrix_shape = ctx->InputShape(0); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| H A D | matrix_solve_op_test.py | 163 def _GenerateTestData(self, matrix_shape, num_rhs): argument
|
| H A D | matrix_solve_ls_op_test.py | 42 def _GenerateTestData(matrix_shape, num_rhs): argument
|
| H A D | tridiagonal_solve_op_test.py | 604 def test_with_matrix_shapes(matrix_shape, rhs_shape=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | xla_hlo_profile_test.cc | 271 Shape matrix_shape = ShapeUtil::MakeShape(F32, {size, size}); in XLA_TEST_F() local
|
| H A D | tuple_test.cc | 171 auto matrix_shape = builder.GetShape(matrix_element).value(); in XLA_TEST_F() local
|
| H A D | while_test.cc | 1176 auto matrix_shape = ShapeUtil::MakeShape(F32, {2, 2}); in XLA_TEST_F() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse_csr.py | 1751 def test_addmv(self, device, dtype, matrix_shape): argument
|