Searched defs:segment_ids_shape (Results 1 – 5 of 5) sorted by relevance
42 …auto segment_ids_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape… in UnsortedSegmentSumInfer() local
34 auto segment_ids_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1); in InitKernel() local
42 auto segment_ids_shape = in_tensors_.at(1)->shape(); in Init() local
237 auto segment_ids_shape = segment_ids->shape()->shape(); in InferImplUnsortedSegmentSum() local292 auto segment_ids_shape = segment_ids->shape()->shape(); in InferImplUnsortedSegmentMax() local344 auto segment_ids_shape = segment_ids->shape()->shape(); in InferImplUnsortedSegmentMin() local