Searched refs:MatrixBandPartOp (Results 1 – 2 of 2) sorted by relevance
46 class MatrixBandPartOp : public OpKernel { class48 explicit MatrixBandPartOp(OpKernelConstruction* context) in MatrixBandPartOp() function in tensorflow::MatrixBandPartOp109 TF_DISALLOW_COPY_AND_ASSIGN(MatrixBandPartOp);115 MatrixBandPartOp<CPUDevice, type>);125 MatrixBandPartOp<CPUDevice, type>);226 MatrixBandPartOp<GPUDevice, type>);241 MatrixBandPartOp<GPUDevice, type>);
26 class MatrixBandPartOp : public XlaOpKernel { class28 explicit MatrixBandPartOp(OpKernelConstruction* context) in MatrixBandPartOp() function in tensorflow::__anon28f8b81c0111::MatrixBandPartOp92 TF_DISALLOW_COPY_AND_ASSIGN(MatrixBandPartOp);94 REGISTER_XLA_OP(Name("MatrixBandPart"), MatrixBandPartOp);