Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SparseSegmentSum.pbtxt37 tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 0]))
41 tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 1]))
46 tf.sparse_segment_sum(c, tf.constant([0, 1, 2]), tf.constant([0, 0, 1]))
/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
Dsegment_reduction_ops_test.py581 ops_list = [(np.add, None, math_ops.sparse_segment_sum),
622 ops_list = [(np.add, None, math_ops.sparse_segment_sum), (
682 math_ops.sparse_segment_sum,
694 ops_list = [(np.add, None, math_ops.sparse_segment_sum), (
709 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
720 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
733 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
746 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
758 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
772 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
[all …]
/external/tensorflow/tensorflow/python/ops/
Dembedding_ops.py577 embeddings = math_ops.sparse_segment_sum(
Dmath_ops.py4679 def sparse_segment_sum(data, function
4750 return gen_math_ops.sparse_segment_sum(
4818 return sparse_segment_sum(
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt2232 name: "sparse_segment_sum"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td16196 tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 0]))
16200 tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 1]))
16205 tf.sparse_segment_sum(c, tf.constant([0, 1, 2]), tf.constant([0, 0, 1]))