MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / NewFillPrimitive

Function NewFillPrimitive

oneflow/user/kernels/same_padding_kernel.cpp:26–29  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

24
25template<typename Context>
26std::unique_ptr<ep::primitive::Fill> NewFillPrimitive(Context* ctx) {
27 const DataType data_type = ctx->TensorDesc4ArgNameAndIndex("y", 0)->data_type();
28 return ep::primitive::NewPrimitive<ep::primitive::FillFactory>(ctx->device_type(), data_type);
29}
30
31template<typename Context>
32std::unique_ptr<ep::primitive::CopyNd> NewCopyNdPrimitive(Context* ctx) {

Callers 2

FillPrimitiveExistsFunction · 0.70
ComputeMethod · 0.70

Calls 3

data_typeMethod · 0.45
device_typeMethod · 0.45

Tested by

no test coverage detected