| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16_grad/ |
| D | unsorted_segment_sum_fp16.cc | 42 auto segment_ids_shape = in_tensors_.at(SECOND_INPUT)->shape(); in Prepare() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32_grad/ |
| D | unsorted_segment_sum.cc | 42 auto segment_ids_shape = in_tensors_.at(1)->shape(); in Prepare() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | unsorted_segment_sum.cc | 40 auto segment_ids_shape = input_args[kInputIndex1]->GetShape()->GetShapeVector(); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | sparse_segment_sum_with_num_segments.cc | 53 auto segment_ids_shape = segment_ids->GetTensorShape(); in Compute() local
|
| D | sparse_segment_mean_with_num_segments.cc | 69 auto segment_ids_shape = segment_ids->GetTensorShape(); in Compute() local
|
| D | sparse_segment_sqrt_n_with_num_segments.cc | 73 auto segment_ids_shape = segment_ids->GetTensorShape(); in Compute() local
|
| D | sparse_segment_sqrt_n.cc | 54 auto segment_ids_shape = segment_ids->GetTensorShape(); in Compute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | sparse_segment_sqrt_n.cc | 39 auto segment_ids_shape = in SparseSegmentSqrtNInferShape() local
|
| D | sparse_segment_sum.cc | 37 auto segment_ids_shape = in SparseSegmentSumInferShape() local
|
| D | sparse_segment_sqrt_n_with_num_segments.cc | 40 auto segment_ids_shape = in SparseSegmentSqrtNWithNumSegmentsInferShape() local
|
| D | sparse_segment_sum_with_num_segments.cc | 35 auto segment_ids_shape = in SparseSegmentSumWithNumSegmentsInferShape() local
|
| D | sparse_segment_mean_with_num_segments.cc | 51 auto segment_ids_shape = in SparseSegmentMeanWithNumSegmentsInferShape() local
|
| D | segment_arithmetic.cc | 80 …auto segment_ids_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->GetShape()… in SegmentArithmeticInferShape() local
|
| D | sparse_segment_mean.cc | 50 …auto segment_ids_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(segment_ids_shape_ptr)[kS… in SparseSegmentMeanInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/ |
| D | sparse_segment_sqrt_n_grad.cc | 52 auto segment_ids_shape = in SparseSegmentSqrtNGradInferShape() local
|
| D | sparse_segment_sum_grad.cc | 52 auto segment_ids_shape = in SparseSegmentSumGradInferShape() local
|
| D | sparse_segment_mean_grad.cc | 55 auto segment_ids_shape = in SparseSegmentMeanGradInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | segment_ops_gpu_kernel.cc | 358 std::vector<int64_t> segment_ids_shape = inputs[1]->GetShapeVector(); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/grad_ops/ |
| D | grad_array_ops.cc | 2560 auto segment_ids_shape = inputs.at(1); in __anonbbaa5cbc1d02() local
|