Searched refs:col_inds (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_matrix_components_op.cc | 82 auto col_inds = col_inds_t->vec<int32>(); in Compute() local
96 /*output*/ col_inds, in Compute()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CSRSparseMatrixComponents.pbtxt | 16 name: "col_inds"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CSRSparseMatrixComponents.pbtxt | 16 name: "col_inds"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CSRSparseMatrixComponents.pbtxt | 17 name: "col_inds"
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| D | csr_sparse_matrix_ops_test.py | 235 self.assertAllEqual(a_csr_mat.indices, a_sm_val.col_inds)
291 self.assertAllEqual(csr_mat.indices, sm_val.col_inds)
367 self.assertAllEqual(a_csr_mat.indices, a_sm_val.col_inds)
416 self.assertAllEqual(csr_mat.indices, sm_val.col_inds)
860 self.assertAllEqual(component.col_inds, np.empty([0], np.int32))
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 38734 …rix tf.Output, index tf.Output, type_ tf.DataType) (row_ptrs tf.Output, col_inds tf.Output, values…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 6570 name: "col_inds"
|