Searched refs:matrix_rank (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | linalg_ops_test.py | 293 self.assertAllEqual([2, 1, 3], self.evaluate(linalg.matrix_rank(x)))
301 linalg.matrix_rank(
321 self.assertAllEqual([2, 1], self.evaluate(linalg.matrix_rank(x)))
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 1268 const int matrix_rank = 2; in CountBatchMatMulOperations() local
1274 a_input.shape(), std::max(matrix_rank, a_input.shape().dim_size()), in CountBatchMatMulOperations()
1277 b_input.shape(), std::max(matrix_rank, b_input.shape().dim_size()), in CountBatchMatMulOperations()
1281 (a_input.shape().dim_size() < matrix_rank) || in CountBatchMatMulOperations()
1282 (b_input.shape().dim_size() < matrix_rank); in CountBatchMatMulOperations()
1296 b_i < bigger_rank_shape->dim_size() - matrix_rank; ++b_i, ++s_i) { in CountBatchMatMulOperations()
1330 for (int i = std::max(0, a_input_shape.dim_size() - matrix_rank); in CountBatchMatMulOperations()
1338 for (int i = std::max(0, b_input_shape.dim_size() - matrix_rank); in CountBatchMatMulOperations()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.linalg.pbtxt | 188 name: "matrix_rank"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.linalg.pbtxt | 200 name: "matrix_rank"
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linalg_impl.py | 748 def matrix_rank(a, tol=None, validate_args=False, name=None): function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 3246 const int matrix_rank = matrix_prop.shape().dim_size(); in ConstantPushDownBiasAdd() local
3256 if (!parent_is_bias_add && child_is_bias_add && const_rank == matrix_rank && in ConstantPushDownBiasAdd()
|