Searched refs:sparse_tensor_dense_matmul (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| H A D | sparse_tensor_dense_matmul_op_test.py | 68 tf_value_ans = sparse_ops.sparse_tensor_dense_matmul(
70 tf_tensor_ans = sparse_ops.sparse_tensor_dense_matmul(
119 result = sparse_ops.sparse_tensor_dense_matmul(x_st, y)
125 result_left_shape_unknown = sparse_ops.sparse_tensor_dense_matmul(
134 sparse_ops.sparse_tensor_dense_matmul(x_st_shape_inconsistent, y)
148 self.evaluate(sparse_ops.sparse_tensor_dense_matmul(sparse_t, dense_t))
151 self.evaluate(sparse_ops.sparse_tensor_dense_matmul(sparse_t, dense_t))
157 sparse_ops.sparse_tensor_dense_matmul(
162 sparse_ops.sparse_tensor_dense_matmul(
175 expected_t, sparse_ops.sparse_tensor_dense_matmul(sparse_t, dense_t))
[all …]
|
| H A D | sparse_tensor_dense_matmul_op_d9m_test.py | 99 result = sparse_ops.sparse_tensor_dense_matmul(
121 result_a = sparse_ops.sparse_tensor_dense_matmul(
124 result_b = sparse_ops.sparse_tensor_dense_matmul(
|
| H A D | sparse_tensor_dense_matmul_grad_test.py | 71 matmul = sparse_ops.sparse_tensor_dense_matmul(
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/ |
| H A D | sparse_tensor_dense_matmul_op_deterministic_test.py | 103 result = sparse_ops.sparse_tensor_dense_matmul(
125 result_a = sparse_ops.sparse_tensor_dense_matmul(
128 result_b = sparse_ops.sparse_tensor_dense_matmul(
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | sparse_ops_test.py | 158 result = sparse_ops.sparse_tensor_dense_matmul(
163 result = sparse_ops.sparse_tensor_dense_matmul(
169 result = sparse_ops.sparse_tensor_dense_matmul(
175 result = sparse_ops.sparse_tensor_dense_matmul(
|
| H A D | sparse_ops.py | 2416 def sparse_tensor_dense_matmul(sp_a, function
2633 sparse_tensor_dense_matmul(b, sp_a, adjoint_a, adjoint_b))
2636 sparse_tensor_dense_matmul(
|
| H A D | array_grad.py | 1067 jac = sparse_ops.sparse_tensor_dense_matmul(sp_mat, grad_flat)
1129 jac = sparse_ops.sparse_tensor_dense_matmul(sp_mat, grad_flat)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | op_callbacks_test.py | 686 y = sparse_ops.sparse_tensor_dense_matmul(sp, w)
698 return sparse_ops.sparse_tensor_dense_matmul(sp, w)
|
| H A D | sparse_tensor_test.py | 89 out = sparse_ops.sparse_tensor_dense_matmul(sp, w)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| H A D | csr_sparse_matrix_ops_test.py | 1430 xw_sparse_tensor = sparse_ops.sparse_tensor_dense_matmul(
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/ |
| H A D | backend.py | 2013 out = sparse_ops.sparse_tensor_dense_matmul(x, y)
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.pbtxt | 2252 name: "sparse_tensor_dense_matmul"
|