Home
last modified time | relevance | path

Searched refs:a_is_sparse (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DSparseMatMul.pbtxt30 name: "a_is_sparse"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DSparseMatMul.pbtxt30 name: "a_is_sparse"
/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
Dsparse_matmul_op_test.py56 a_is_sparse=sp_a,
151 a_is_sparse=sp_a,
Dsparse_tensor_dense_matmul_op_test.py267 a_is_sparse=True,
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2021-044.md18 a_is_sparse=True, b_is_sparse=True)
Dtfsa-2021-178.md17 a_is_sparse=False,
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.linalg.pbtxt189 …b\', \'transpose_a\', \'transpose_b\', \'adjoint_a\', \'adjoint_b\', \'a_is_sparse\', \'b_is_spars…
201 …argspec: "args=[\'a\', \'b\', \'transpose_a\', \'adjoint_a\', \'a_is_sparse\', \'b_is_sparse\', \'…
Dtensorflow.pbtxt1645 …b\', \'transpose_a\', \'transpose_b\', \'adjoint_a\', \'adjoint_b\', \'a_is_sparse\', \'b_is_spars…
2173 …argspec: "args=[\'a\', \'b\', \'transpose_a\', \'transpose_b\', \'a_is_sparse\', \'b_is_sparse\', …
Dtensorflow.raw_ops.pbtxt4373 …argspec: "args=[\'a\', \'b\', \'transpose_a\', \'transpose_b\', \'a_is_sparse\', \'b_is_sparse\', …
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.linalg.pbtxt201 …b\', \'transpose_a\', \'transpose_b\', \'adjoint_a\', \'adjoint_b\', \'a_is_sparse\', \'b_is_spars…
213 …argspec: "args=[\'a\', \'b\', \'transpose_a\', \'adjoint_a\', \'a_is_sparse\', \'b_is_sparse\', \'…
Dtensorflow.pbtxt809 …b\', \'transpose_a\', \'transpose_b\', \'adjoint_a\', \'adjoint_b\', \'a_is_sparse\', \'b_is_spars…
Dtensorflow.raw_ops.pbtxt4373 …argspec: "args=[\'a\', \'b\', \'transpose_a\', \'transpose_b\', \'a_is_sparse\', \'b_is_sparse\', …
/external/tensorflow/tensorflow/python/ops/
Dmath_ops.py3497 a_is_sparse=False, argument
3653 if (not a_is_sparse and
3681 if a_is_sparse or b_is_sparse:
3698 a_is_sparse=a_is_sparse,
3724 a_is_sparse=False, argument
3813 a_is_sparse=a_is_sparse,
Dmath_grad.py1777 a_is_sparse=t1_sparse,
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_math_ops.py797 a_is_sparse=False, argument
840 a_is_sparse=a_is_sparse,
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator.py1481 a_is_sparse=False, argument
1487 if a_is_sparse or b_is_sparse:
/external/tensorflow/tensorflow/compiler/tests/
Dbinary_ops_test.py1060 return math_ops.sparse_matmul(a, b, a_is_sparse=True)
1066 return math_ops.sparse_matmul(a, b, a_is_sparse=True, b_is_sparse=True)
/external/pytorch/test/
Dtest_autograd.py2335 for a_req_grad, a_is_sparse, b_req_grad, b_is_sparse in mm_test_cases:
2338 if not ((a_is_sparse or b_is_sparse) and (a_req_grad or b_req_grad)):
2341 if a_is_sparse:
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-no-tf2xla-fallback.mlir1394 …%0 = "tf.SparseMatMul"(%arg0, %arg1) {a_is_sparse = true, b_is_sparse = false, transpose_a = false…
1411 …%0 = "tf.SparseMatMul"(%arg0, %arg1) {a_is_sparse = true, b_is_sparse = false, transpose_a = false…
1427 …%0 = "tf.SparseMatMul"(%arg0, %arg1) {a_is_sparse = true, b_is_sparse = false, transpose_a = false…
Dlegalize-tf.mlir1411 …%0 = "tf.SparseMatMul"(%arg0, %arg1) {a_is_sparse = true, b_is_sparse = false, transpose_a = false…
1428 …%0 = "tf.SparseMatMul"(%arg0, %arg1) {a_is_sparse = true, b_is_sparse = false, transpose_a = false…
1444 …%0 = "tf.SparseMatMul"(%arg0, %arg1) {a_is_sparse = true, b_is_sparse = false, transpose_a = false…
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt33829 name: "a_is_sparse"
33872 name: "a_is_sparse"
Dops_history.v2.pbtxt75509 name: "a_is_sparse"
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt52505 name: "a_is_sparse"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td15926 DefaultValuedAttr<BoolAttr, "false">:$a_is_sparse,