| /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 | 779 class SparseMatMul { class
855 TF_DISALLOW_COPY_AND_ASSIGN(SparseMatMul);
884 typename SparseMatMul<TL, TR>::TensorInfoCache
1085 inline void SparseMatMul<TL, TR>::ComputeOutputBlock( in ComputeOutputBlock()
1087 const typename SparseMatMul<TL, TR>::ConstMatrixMapR& right, int num_cols, in ComputeOutputBlock()
1137 SparseMatMul<TL, TR>::CreateSparseSlices( in CreateSparseSlices()
1138 const typename SparseMatMul<TL, TR>::ConstMatrixMapL& mat, bool transpose, in CreateSparseSlices()
1152 SparseMatMul<TL, TR>::ConstMatrixMapL* slice, in CreateSparseSlices()
1169 SparseMatMul<TL, TR>::ConstMatrixMapL* slice = nullptr; in CreateSparseSlices()
1171 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
114 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/ops/compat/ops_history_v2/ |
| D | SparseMatMul.pbtxt | 2 name: "SparseMatMul"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SparseMatMul.pbtxt | 2 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/core/grappler/optimizers/ |
| D | auto_mixed_precision_test.cc | 295 Output deny2 = ops::SparseMatMul(s.WithOpName("deny2"), clr4, clr4); in TEST_F()
1243 Output deny3 = ops::SparseMatMul(s.WithOpName("deny3"), clr4, clr4); in TEST_F()
|
| D | arithmetic_optimizer_test.cc | 731 matmul = ops::SparseMatMul(matmul_op, trans_a, trans_b); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 1198 // Verify SparseMatMul is legalized to dot.
1202 …%0 = "tf.SparseMatMul"(%arg0, %arg1) {a_is_sparse = true, b_is_sparse = false, transpose_a = false…
1206 // SparseMatMul where one operand needs to be transposed and the other one not.
1219 …%0 = "tf.SparseMatMul"(%arg0, %arg1) {a_is_sparse = true, b_is_sparse = false, transpose_a = false…
1223 // SparseMatMul where one operand needs to be casted and the other one not.
1235 …%0 = "tf.SparseMatMul"(%arg0, %arg1) {a_is_sparse = true, b_is_sparse = false, transpose_a = false…
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 3053 TEST_F(OpTest, SparseMatMul) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 2779 ### tf.SparseMatMul
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 4308 name: "SparseMatMul"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 4308 name: "SparseMatMul"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 33801 name: "SparseMatMul"
33844 name: "SparseMatMul"
|
| D | ops_history.v2.pbtxt | 75506 name: "SparseMatMul"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 24919 func SparseMatMul(scope *Scope, a tf.Output, b tf.Output, optional ...SparseMatMulAttr) (product tf… func
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 14274 def TF_SparseMatMulOp : TF_Op<"SparseMatMul", [NoSideEffect]> {
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 48849 name: "SparseMatMul"
|