Home
last modified time | relevance | path

Searched defs:matrix_shape (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatrix_solve_op.cc32 const TensorShape matrix_shape = ctx->InputShape(0); in Compile() local
/external/tensorflow/tensorflow/core/ops/
Dsparse_csr_matrix_ops.cc45 const ShapeHandle& matrix_shape, in ValidateSquareMatrixShape()
558 ShapeHandle matrix_shape = sparse_matrix_shape_and_type.shape; in __anon59a9d6c30f02() local
580 ShapeHandle matrix_shape = sparse_matrix_shape_and_type.shape; in __anon59a9d6c31002() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dmatrix_solve_op_test.py161 def _GenerateTestData(self, matrix_shape, num_rhs): argument
Dmatrix_solve_ls_op_test.py46 def _GenerateTestData(matrix_shape, num_rhs): argument
Dtridiagonal_solve_op_test.py611 def test_with_matrix_shapes(matrix_shape, rhs_shape=None): argument
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc106 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
315 Shape matrix_shape = ShapeUtil::MakeShape(F32, {8, 8}); in TEST_F() local
1271 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
1280 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}, {true, true}); in TEST_F() local
1290 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
1299 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
1308 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
1317 Shape matrix_shape = ShapeUtil::MakeShape(F32, {128, 64}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dxla_hlo_profile_test.cc270 Shape matrix_shape = ShapeUtil::MakeShape(F32, {size, size}); in XLA_TEST_F() local
Dtuple_test.cc171 auto matrix_shape = builder.GetShape(matrix_element).ConsumeValueOrDie(); in XLA_TEST_F() local
Dwhile_test.cc1190 auto matrix_shape = ShapeUtil::MakeShape(F32, {2, 2}); in XLA_TEST_F() local