| /external/tensorflow/tensorflow/core/kernels/ |
| segment_reduction_ops_gpu.cu.cc | 54 const Index* segment_ids,
63 Index first_segment_id = segment_ids[input_outer_dim_index_base];
71 segment_ids[input_outer_dim_index_base + j];
103 // 'segment_ids' mapping and 'inner_dim_size'.
108 const Index* segment_ids,
115 const Index output_segment_index = segment_ids[input_segment_index];
131 typename TTypes<Index>::ConstFlat segment_ids, const Index data_size,
147 // *) 'segment_ids.shape' is a prefix of data's shape.
150 const Index input_outer_dim_size = segment_ids.dimension(0);
165 segment_ids.data(), data, output.data(), total_stripe_count)
[all...] |
| segment_reduction_ops.cc | 54 const Tensor& segment_ids) {
55 OP_REQUIRES(context, TensorShapeUtils::IsVector(segment_ids.shape()),
56 errors::InvalidArgument("segment_ids should be a vector."));
57 const int64 num_indices = segment_ids.NumElements();
60 "segment_ids should be the same size as dimension 0 of"
66 const Tensor& segment_ids) {
67 SegmentReductionValidationHelper(c, input, segment_ids);
82 const Tensor& segment_ids = context->input(1); variable
84 if (!SegmentReductionDoValidation(context, input, segment_ids)) {
88 const int64 num_indices = segment_ids.NumElements()
220 const Tensor& segment_ids = context->input(1); variable
470 const Tensor& segment_ids = context->input(1); variable
629 const Tensor& segment_ids = context->input(2); variable
1005 const Tensor& segment_ids = context->input(2); variable
[all...] |
| segment_reduction_ops.h | 34 // 'segment_ids').
35 // segment_ids_shape: shape of 'segment_ids' tensor.
36 // segment_ids: unsorted map from input to output segment ids at which to
45 typename TTypes<Index>::ConstFlat segment_ids,
57 typename TTypes<Index>::ConstFlat segment_ids,
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| embedding_ops.py | 423 segment_ids = sparse_values.indices[:, 0]
424 if segment_ids.dtype != dtypes.int32:
425 segment_ids = math_ops.cast(segment_ids, dtypes.int32)
434 embeddings = math_ops.sparse_segment_sum(embeddings, idx, segment_ids,
437 embeddings = math_ops.sparse_segment_mean(embeddings, idx, segment_ids,
440 embeddings = math_ops.sparse_segment_sqrt_n(embeddings, idx, segment_ids,
540 segment_ids = sp_values.indices[:, 0]
544 sampled_candidates = array_ops.gather(sampled_candidates, segment_ids)
557 if segment_ids.dtype != dtypes.int32
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| embedding_ops.py | 434 segment_ids = sp_ids.indices[:, 0]
435 if segment_ids.dtype != dtypes.int32:
436 segment_ids = math_ops.cast(segment_ids, dtypes.int32)
470 embeddings = math_ops.segment_sum(embeddings, segment_ids, name=name)
472 embeddings = math_ops.segment_sum(embeddings, segment_ids)
473 weight_sum = math_ops.segment_sum(weights, segment_ids)
476 embeddings = math_ops.segment_sum(embeddings, segment_ids)
478 weight_sum = math_ops.segment_sum(weights_squared, segment_ids)
487 embeddings, idx, segment_ids, name=name
[all...] |
| math_grad_test.py | 160 segment_ids = constant_op.constant([0, 0, 1], dtype=dtypes.int64)
161 segment_min = math_ops.segment_min(data, segment_ids)
169 segment_ids = constant_op.constant([0, 0, 1], dtype=dtypes.int64)
170 segment_max = math_ops.segment_max(data, segment_ids)
179 segment_ids = constant_op.constant([0, 0], dtype=dtypes.int64)
180 segment_min = math_ops.segment_min(data, segment_ids)
189 segment_ids = constant_op.constant([0, 0], dtype=dtypes.int64)
190 segment_max = math_ops.segment_max(data, segment_ids)
|
| math_ops.py | [all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| segment_reduction_ops_test.py | 118 s = tf_op(data=tf_x, segment_ids=indices)
132 math_ops.segment_sum(data=tf_x, segment_ids=indices)
140 s = math_ops.segment_sum(data=tf_x, segment_ids=indices)
141 with self.assertRaisesOpError("segment_ids should be the same size"):
151 result = math_ops.segment_sum(data=tf_x, segment_ids=indices).eval()
161 s = math_ops.segment_sum(data=tf_x, segment_ids=indices)
172 s = math_ops.segment_sum(data=tf_x, segment_ids=indices)
181 s = math_ops.segment_sum(data=tf_x, segment_ids=indices)
184 "'segment_ids' input is not sorted."):
192 s = math_ops.segment_sum(data=tf_x, segment_ids=indices
[all...] |
| /external/tensorflow/tensorflow/contrib/signal/python/ops/ |
| reconstruction_ops.py | 133 segment_ids = array_ops.reshape(shape_ops.frame(
136 result = math_ops.unsorted_segment_sum(subframe_signal, segment_ids,
|
| /external/libvpx/libvpx/vp9/common/ |
| vp9_pred_common.h | 23 const uint8_t *segment_ids, BLOCK_SIZE bsize,
35 VPXMIN(segment_id, segment_ids[mi_offset + y * cm->mi_cols + x]);
|
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
| /external/libvpx/libvpx/vp9/decoder/ |
| vp9_decodemv.c | 91 static int dec_get_segment_id(const VP9_COMMON *cm, const uint8_t *segment_ids,
98 VPXMIN(segment_id, segment_ids[mi_offset + y * cm->mi_cols + x]);
|
