| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_impl_1.cc | 24 const Tensor& segment_ids) { in SegmentReductionValidationHelper()
37 const Tensor& segment_ids) { in SegmentReductionDoValidation()
46 const Tensor& segment_ids, in UnsortedSegmentReductionValidation()
62 const Tensor& segment_ids, in UnsortedSegmentReductionDoValidation()
|
| D | segment_reduction_ops_gpu.cu.cc | 58 const Index output_outer_dim_size, const Index* __restrict__ segment_ids, in SortedSegmentReductionCustomKernel()
112 const int64 output_outer_dim_size, const Index* __restrict__ segment_ids, in UnsortedSegmentCustomKernel()
138 typename TTypes<Index>::ConstFlat segment_ids, in operator ()()
|
| D | segment_reduction_ops_impl.h | 93 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_math_ops.py | 161 segment_ids, argument
264 def segment_sum(data, segment_ids, num_segments, name=None): argument
274 def segment_prod(data, segment_ids, num_segments, name=None): argument
284 def segment_min(data, segment_ids, num_segments, name=None): argument
294 def segment_max(data, segment_ids, num_segments, name=None): argument
304 def segment_mean(data, segment_ids, num_segments, name=None): argument
320 def segment_sqrt_n(data, segment_ids, num_segments, name=None): argument
|
| 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 | 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::__anonb8ea79160111::SegmentSumOpModel
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_ops.py | 4176 def _unsorted_segment_N(data, segment_ids, num_segments): argument
4204 def unsorted_segment_mean(data, segment_ids, num_segments, name=None): argument
4250 def unsorted_segment_sqrt_n(data, segment_ids, num_segments, name=None): argument
4297 segment_ids, argument
4373 segment_ids, argument
4442 segment_ids, argument
4488 segment_ids, argument
4527 segment_ids, argument
4564 segment_ids, argument
|
| D | math_grad_test.py | 380 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
|
| /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 | 4177 Output segment_ids = in TEST_F() local
4226 Output segment_ids = ops::Cast( in TEST_F() local
|
| /external/libaom/libaom/av1/encoder/ |
| D | bitstream.c | 403 const CommonModeInfoParams *const mi_params, uint8_t *segment_ids, in set_spatial_segment_id()
|