| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_segment_op_test.py | 55 def expected_value(self, data, segment_ids, num_segments, combiner): argument
105 def testRaggedSegment_Int(self, segment_op, combiner, segment_ids): argument
141 def testRaggedSegment_Float(self, segment_op, combiner, segment_ids): argument
|
| D | ragged_math_ops.py | 151 segment_ids, argument
245 def segment_sum(data, segment_ids, num_segments, name=None): argument
252 def segment_prod(data, segment_ids, num_segments, name=None): argument
259 def segment_min(data, segment_ids, num_segments, name=None): argument
266 def segment_max(data, segment_ids, num_segments, name=None): argument
273 def segment_mean(data, segment_ids, num_segments, name=None): argument
287 def segment_sqrt_n(data, segment_ids, num_segments, name=None): argument
|
| D | segment_id_ops.py | 69 def segment_ids_to_row_splits(segment_ids, num_segments=None, name=None): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_gpu.cu.cc | 57 const Index* segment_ids, in SortedSegmentSumCustomKernel()
111 const Index* segment_ids, in UnsortedSegmentCustomKernel()
134 typename TTypes<Index>::ConstFlat segment_ids, const Index data_size, in operator ()()
|
| D | segment_reduction_ops.cc | 56 const Tensor& segment_ids) { in SegmentReductionValidationHelper()
68 const Tensor& segment_ids) { in SegmentReductionDoValidation()
84 const Tensor& segment_ids = context->input(1); in Compute() local
222 const Tensor& segment_ids = context->input(1); in ComputeAsync() local
439 const Tensor& segment_ids, in UnsortedSegmentReductionValidation()
455 const Tensor& segment_ids, in UnsortedSegmentReductionDoValidation()
474 const Tensor& segment_ids = context->input(1); in Compute() local
633 const Tensor& segment_ids = context->input(2); in Compute() local
1014 const Tensor& segment_ids = context->input(2); in Compute() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_ops.py | 3208 def _unsorted_segment_N(data, segment_ids, num_segments): argument
3228 def unsorted_segment_mean(data, segment_ids, num_segments, name=None): argument
3274 def unsorted_segment_sqrt_n(data, segment_ids, num_segments, name=None): argument
3320 def sparse_segment_sum(data, indices, segment_ids, name=None, argument
3395 segment_ids, argument
3406 segment_ids, argument
3451 segment_ids, argument
3489 segment_ids, argument
3526 segment_ids, argument
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops.py | 722 segment_ids=None): argument
|
| /external/libvpx/libvpx/vp9/common/ |
| D | vp9_pred_common.h | 23 const uint8_t *segment_ids, BLOCK_SIZE bsize, in get_segment_id()
|
| /external/libaom/libaom/av1/common/ |
| D | pred_common.h | 25 const uint8_t *segment_ids, BLOCK_SIZE bsize, in get_segment_id()
|
| /external/libvpx/libvpx/vp9/decoder/ |
| D | vp9_decodemv.c | 91 static int dec_get_segment_id(const VP9_COMMON *cm, const uint8_t *segment_ids, in dec_get_segment_id()
|
| /external/libaom/libaom/av1/decoder/ |
| D | decodemv.c | 283 static int dec_get_segment_id(const AV1_COMMON *cm, const uint8_t *segment_ids, in dec_get_segment_id()
|
| /external/libaom/libaom/av1/encoder/ |
| D | bitstream.c | 415 uint8_t *segment_ids, in set_spatial_segment_id()
|