Home
last modified time | relevance | path

Searched defs:input_shape_b (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/
Dmasked_select_grad.cc99 auto input_shape_b = ctx.Input(1)->GetTensorShape()->GetDimSizes(); in MaskedSelectGradCompute() local
Dselect.cc82 auto input_shape_b = ctx.Input(2)->GetTensorShape()->GetDimSizes(); in SelectParamCheck() local
Dmasked_scatter.cc95 auto input_shape_b = ctx.Input(1)->GetTensorShape(); in InputCheck() local
Dmasked_select.cc257 auto input_shape_b = ctx.Input(1)->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 …erator::BroadcastIterator(std::vector<int64_t> &input_shape_a, std::vector<int64_t> &input_shape_b, 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()