Home
last modified time | relevance | path

Searched defs:bc_axis (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Ddynamic_broadcast_gradient_args.cc136 ShapeArray bc_axis; in BroadcastGradientArgsInferValue() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/grad_ops/
Dcommon_utils.cc277 …std::vector<std::vector<int64_t>> bc_axis = BroadcastGradientArgsInferValue(broadcast_shape[0], br… in StaticBinopGradCommon() local
292 std::vector<std::vector<int64_t>> bc_axis = in MatMulExtBroadCastGradPart() local
Dgrad_array_ops.cc751 ShapeArray bc_axis; in BroadcastGradientArgsInferValueSelect() local
851 std::vector<std::vector<int64_t>> bc_axis = BroadcastGradientArgsInferValueSelect( in StaticBinopGradSelect() local