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/
Dsegment_reduction_ops_test.py496 ops_list = [(np.add, None, math_ops.sparse_segment_sum),
527 ops_list = [(np.add, None, math_ops.sparse_segment_sum), (
585 ops_list = [(np.add, None, math_ops.sparse_segment_sum), (
600 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
611 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
624 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
637 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
649 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
663 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
677 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
[all …]
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dembedding_ops.py448 embeddings = math_ops.sparse_segment_sum(embeddings, idx, segment_ids,
769 return math_ops.sparse_segment_sum(ret, idx, segment_ids, name=name)
915 partitioned_result[p] = math_ops.sparse_segment_sum(
/external/tensorflow/tensorflow/python/ops/
Dembedding_ops.py531 embeddings = math_ops.sparse_segment_sum(
Dmath_ops.py3320 def sparse_segment_sum(data, indices, segment_ids, name=None, function
3388 return gen_math_ops.sparse_segment_sum(
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt2140 name: "sparse_segment_sum"