Searched refs:FillOp (Results 1 – 2 of 2) sorted by relevance
30 class FillOp : public XlaOpKernel { class32 explicit FillOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in FillOp() function in tensorflow::__anonfe8d977b0111::FillOp55 REGISTER_XLA_OP(Name("Fill").CompileTimeConstantInput("dims"), FillOp);
151 class FillOp : public OpKernel { class153 explicit FillOp(OpKernelConstruction* context) : OpKernel(context) {} in FillOp() function in tensorflow::FillOp183 FillOp<D##Device, TYPE, int32>); \189 FillOp<D##Device, TYPE, int64>);215 FillOp<CPUDevice, int32, int32>);244 FillOp<CPUDevice, int32, int32>);