Searched refs:num_segments_value (Results 1 – 2 of 2) sorted by relevance
317 int64_t num_segments_value = 0; in GetUnsortedSegmentOpScalarArg() local327 num_segments_value = *static_cast<int64_t *>(num_segments_tensor->data_c()); in GetUnsortedSegmentOpScalarArg()329 num_segments_value = *static_cast<int32_t *>(num_segments_tensor->data_c()); in GetUnsortedSegmentOpScalarArg()334 num_segments_value = GetValue<int64_t>(num_segments->BuildValue()); in GetUnsortedSegmentOpScalarArg()336 num_segments_value = GetValue<int32_t>(num_segments->BuildValue()); in GetUnsortedSegmentOpScalarArg()341 return num_segments_value; in GetUnsortedSegmentOpScalarArg()
245 int64_t num_segments_value = GetUnsortedSegmentOpScalarArg(args_spec_list, op_name); in InferImplUnsortedSegmentSum() local246 if (num_segments_value <= 0) { in InferImplUnsortedSegmentSum()249 shape.emplace_back(num_segments_value); in InferImplUnsortedSegmentSum()261 min_shape.emplace_back(num_segments_value); in InferImplUnsortedSegmentSum()262 max_shape.emplace_back(num_segments_value); in InferImplUnsortedSegmentSum()300 int64_t num_segments_value = GetUnsortedSegmentOpScalarArg(args_spec_list, op_name); in InferImplUnsortedSegmentMax() local301 if (num_segments_value <= 0) { in InferImplUnsortedSegmentMax()304 shape.emplace_back(num_segments_value); in InferImplUnsortedSegmentMax()314 min_shape.emplace_back(num_segments_value); in InferImplUnsortedSegmentMax()315 max_shape.emplace_back(num_segments_value); in InferImplUnsortedSegmentMax()[all …]