Home
last modified time | relevance | path

Searched refs:sparse_reduce_sum (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
H A Dsparse_ops_test.py715 tf_dense_ans = sparse_ops.sparse_reduce_sum(sp_t, reduction_axes,
774 self.evaluate(sparse_ops.sparse_reduce_sum(sp_t, -3))
776 self.evaluate(sparse_ops.sparse_reduce_sum(sp_t, 2))
792 reduced = sparse_ops.sparse_reduce_sum(sp_t, axes)
800 reduced = sparse_ops.sparse_reduce_sum(sp_t, -1)
811 tf_op = sparse_ops.sparse_reduce_sum
871 res = sparse_ops.gen_sparse_ops.sparse_reduce_sum(
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/tests/
H A Dgen_sparse_tensor_input.py43 r = sparse_ops.sparse_reduce_sum(x)
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dsparse_ops.py1533 return gen_sparse_ops.sparse_reduce_sum(
1549 def sparse_reduce_sum(sp_input, axis=None, keepdims=None, function
1606 return gen_sparse_ops.sparse_reduce_sum(
H A Dsparse_grad.py309 sum_reduced = -sparse_ops.sparse_reduce_sum(sp_product, [-1], keepdims=True)
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.pbtxt2200 name: "sparse_reduce_sum"
/aosp_15_r20/external/tensorflow/
H A DRELEASE.md9457 `tf.sparse_reduce_sum`: `reduction_axes` becomes `axis` *