Searched refs:TileOp (Results 1 – 2 of 2) sorted by relevance
39 class TileOp : public XlaOpKernel { class41 explicit TileOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in TileOp() function in tensorflow::__anona5aa93550111::TileOp111 TF_DISALLOW_COPY_AND_ASSIGN(TileOp);114 REGISTER_XLA_OP(Name("Tile").CompileTimeConstantInput("multiples"), TileOp);
85 class TileOp : public OpKernel { class87 explicit TileOp(OpKernelConstruction* context) : OpKernel(context) {} in TileOp() function in tensorflow::TileOp178 TF_DISALLOW_COPY_AND_ASSIGN(TileOp);183 inline void TileOp<Device, Tmultiples>::HandleCase( in HandleCase()196 void TileOp<device, Tmultiples>::HandleCase<dtype>( \527 TileOp<CPUDevice, int32>);532 TileOp<CPUDevice, int64>);551 TileOp<GPUDevice, int32>); \557 TileOp<GPUDevice, int64>);599 TileOp<SYCLDevice, int32>); \[all …]