| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_gpu.cu.h | 63 const Index output_outer_dim_size, const Index* __restrict__ segment_ids, in SortedSegmentReductionCustomKernel()
117 const int64 output_outer_dim_size, const Index* __restrict__ segment_ids, in UnsortedSegmentCustomKernel()
137 const Tsegmentids* __restrict__ segment_ids, // [size] in SegmentOffsetsKernel()
167 const Tsegmentids* segment_ids, // [size] in LaunchSegmentOffsetsKernel()
|
| D | segment_reduction_ops_impl_1.cc | 23 const Tensor& segment_ids) { in ValidateSegmentReduction()
44 const Tensor& segment_ids, in ValidateUnsortedSegmentReduction()
64 const Tensor& segment_ids, in ValidateSparseSegmentReduction()
|
| D | segment_reduction_ops_impl.h | 90 const Tensor& segment_ids = context->input(1); in Compute() local
233 const Tensor& segment_ids = context->input(1); in ComputeAsync() local
|
| /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 | 162 segment_ids, argument
265 def segment_sum(data, segment_ids, num_segments, name=None): argument
275 def segment_prod(data, segment_ids, num_segments, name=None): argument
285 def segment_min(data, segment_ids, num_segments, name=None): argument
295 def segment_max(data, segment_ids, num_segments, name=None): argument
305 def segment_mean(data, segment_ids, num_segments, name=None): argument
321 def segment_sqrt_n(data, segment_ids, num_segments, name=None): argument
|
| D | segment_id_ops.py | 80 def segment_ids_to_row_splits(segment_ids, num_segments=None, argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | segment_sum.cc | 35 const TfLiteTensor* segment_ids, in ResizeOutputTensor()
70 const TfLiteTensor* segment_ids; in Prepare() local
92 const TfLiteTensor* segment_ids; in Eval() local
|
| D | segment_sum_test.cc | 31 SegmentSumOpModel(const TensorData& data, const TensorData& segment_ids) { in SegmentSumOpModel()
40 int segment_ids() const { return segment_ids_id_; } in segment_ids() function in tflite::__anon5aa192bb0111::SegmentSumOpModel
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_ops.py | 4484 def _unsorted_segment_N(data, segment_ids, num_segments): argument
4512 def unsorted_segment_mean(data, segment_ids, num_segments, name=None): argument
4558 def unsorted_segment_sqrt_n(data, segment_ids, num_segments, name=None): argument
4605 segment_ids, argument
4681 segment_ids, argument
4750 segment_ids, argument
4796 segment_ids, argument
4835 segment_ids, argument
4872 segment_ids, argument
|
| D | math_grad_test.py | 379 def _run_gradient_check(self, data, segment_ids): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | unsorted_segment_join_op_test.py | 190 def test_multiple_cases_with_different_dims(self, inputs, segment_ids, argument
|
| D | segment_reduction_ops_test.py | 522 def _sparseSegmentReduceGrad(self, ygrad, indices, segment_ids, output_dim0, 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/tflite-support/tensorflow_lite_support/cc/task/text/qa/ |
| D | bert_question_answerer.cc | 230 std::vector<int> segment_ids; in Preprocess() local
|
| /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 | 300 static int dec_get_segment_id(const AV1_COMMON *cm, const uint8_t *segment_ids, in dec_get_segment_id()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 4634 Output segment_ids = in TEST_F() local
4678 Output segment_ids = in TEST_F() local
4755 Output segment_ids = ops::Cast( in TEST_F() local
|
| /external/libaom/libaom/av1/encoder/ |
| D | bitstream.c | 416 const CommonModeInfoParams *const mi_params, uint8_t *segment_ids, in set_spatial_segment_id()
|