Home
last modified time | relevance | path

Searched defs:input_shape_a (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dselect.cc81 auto input_shape_a = ctx.Input(1)->GetTensorShape()->GetDimSizes(); in SelectParamCheck() local
114 auto input_shape_a = ctx.Input(1)->GetTensorShape()->GetDimSizes(); in SelectCompute() local
Dmasked_select_grad.cc98 auto input_shape_a = ctx.Input(0)->GetTensorShape()->GetDimSizes(); in MaskedSelectGradCompute() local
Dmasked_scatter.cc94 auto input_shape_a = ctx.Input(0)->GetTensorShape(); in InputCheck() local
Dmasked_select.cc256 auto input_shape_a = ctx.Input(0)->GetTensorShape()->GetDimSizes(); in MaskedSelectCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/utils/
Dbroadcast_iterator.cc22 BroadcastIterator::BroadcastIterator(std::vector<int64_t> &input_shape_a, std::vector<int64_t> &inp… in BroadcastIterator()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dcpu_kernel.cc315 BroadcastIterator::BroadcastIterator(ShapeVector input_shape_a, ShapeVector input_shape_b, ShapeVec… in BroadcastIterator()