Searched refs:nnz_a (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_add_op_test.py | 149 sp_a, nnz_a = self._randomTensor([n, m], np.float32)
155 [sp_a.values, sp_b.values], [(nnz_a,), (nnz_b,)], sp_sum.values,
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 123 OpContext DescribeSparseTensorDenseMatMul(const int nnz_a, in DescribeSparseTensorDenseMatMul() argument
130 DescribeArbitraryRankInput({nnz_a, 2}, DT_INT64, &op_context.op_info); in DescribeSparseTensorDenseMatMul()
131 DescribeArbitraryRankInput({nnz_a}, DT_FLOAT, &op_context.op_info); in DescribeSparseTensorDenseMatMul()
|