Searched refs:SparseMatMul (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_SparseMatMul.pbtxt | 2 graph_op_name: "SparseMatMul"
4 name: "sparse.SparseMatMul"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 780 class SparseMatMul { class
856 TF_DISALLOW_COPY_AND_ASSIGN(SparseMatMul);
885 typename SparseMatMul<TL, TR>::TensorInfoCache
1086 inline void SparseMatMul<TL, TR>::ComputeOutputBlock( in ComputeOutputBlock()
1088 const typename SparseMatMul<TL, TR>::ConstMatrixMapR& right, int num_cols, in ComputeOutputBlock()
1138 SparseMatMul<TL, TR>::CreateSparseSlices( in CreateSparseSlices()
1139 const typename SparseMatMul<TL, TR>::ConstMatrixMapL& mat, bool transpose, in CreateSparseSlices()
1153 SparseMatMul<TL, TR>::ConstMatrixMapL* slice, in CreateSparseSlices()
1170 SparseMatMul<TL, TR>::ConstMatrixMapL* slice = nullptr; in CreateSparseSlices()
1172 slice = new SparseMatMul<TL, TR>::ConstMatrixMapL( in CreateSparseSlices()
[all …]
|
| D | sparse_matmul_op_test.cc | 89 static Graph* SparseMatMul(int m, int n, int d, float sparsity_a, in SparseMatMul() function
117 auto g = SparseMatMul<TA, TB>(M, N, K, S1 / 100.0, S2 / 100.0, TRA, TRB); \
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_SparseMatMul.pbtxt | 2 graph_op_name: "SparseMatMul"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseMatMul.pbtxt | 2 graph_op_name: "SparseMatMul"
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | gpu_supported_ops.md | 221 `SparseMatMul` | `Tb={float}`<br>`Ta={float}`
|
| D | cpu_supported_ops.md | 224 `SparseMatMul` | `Tb={float}`<br>`Ta={float}`
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 3052 TEST_F(OpTest, SparseMatMul) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 804 Output matmul = ops::SparseMatMul(matmul_op, trans_a, trans_b); in TEST_F()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3496 name: "SparseMatMul"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3496 name: "SparseMatMul"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 33801 name: "SparseMatMul"
33844 name: "SparseMatMul"
|
| D | ops_history.v1.pbtxt | 76796 name: "SparseMatMul"
|
| D | ops_history.v2.pbtxt | 76169 name: "SparseMatMul"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 14653 func SparseMatMul(scope *Scope, a tf.Output, b tf.Output, optional ...SparseMatMulAttr) (product tf… func
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 36573 name: "SparseMatMul"
|