Searched refs:MaskedFill (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | masked_fill.h | 32 class MaskedFill : public PrimitiveC {
35 MaskedFill() : PrimitiveC(kNameMaskedFill) { InitIOName({"input", "mask", "value"}, {"output"}); } in MaskedFill() function
37 ~MaskedFill() = default;
38 MS_DECLARE_PARENT(MaskedFill, PrimitiveC);
43 using PrimMaskedFillPtr = std::shared_ptr<MaskedFill>;
|
| D | masked_fill.cc | 77 REGISTER_PRIMITIVE_EVAL_IMPL(MaskedFill, prim::kPrimMaskedFill, MaskedFillInfer, nullptr, true);
|
| /third_party/mindspore/mindspore/ops/_grad_experimental/ |
| D | grad_array_ops.py | 26 @bprop_getters.register(P.MaskedFill)
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 36 … EmbeddingLookup, Unique, GatherD, Identity, Range, MaskedFill, MaskedSelect, SearchSorted,
|
| D | array_ops.py | 6045 class MaskedFill(Primitive): class
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_ops.py | 2575 'block': P.MaskedFill(),
|