Searched refs:MatrixSetDiagOp (Results 1 – 3 of 3) sorted by relevance
44 class MatrixSetDiagOp : public OpKernel { class46 explicit MatrixSetDiagOp(OpKernelConstruction* context) : OpKernel(context) { in MatrixSetDiagOp() function in tensorflow::MatrixSetDiagOp173 TF_DISALLOW_COPY_AND_ASSIGN(MatrixSetDiagOp);179 MatrixSetDiagOp<CPUDevice, type>); \182 MatrixSetDiagOp<CPUDevice, type>); \185 MatrixSetDiagOp<CPUDevice, type>);195 MatrixSetDiagOp<CPUDevice, type>);283 MatrixSetDiagOp<GPUDevice, type>); \288 MatrixSetDiagOp<GPUDevice, type>); \293 MatrixSetDiagOp<GPUDevice, type>);[all …]
454 class MatrixSetDiagOp : public XlaOpKernel { class456 explicit MatrixSetDiagOp(OpKernelConstruction* context) in MatrixSetDiagOp() function in tensorflow::MatrixSetDiagOp535 TF_DISALLOW_COPY_AND_ASSIGN(MatrixSetDiagOp);538 REGISTER_XLA_OP(Name("MatrixSetDiag"), MatrixSetDiagOp);540 MatrixSetDiagOp);542 MatrixSetDiagOp);
2692 void MatrixSetDiagOp::getCanonicalizationPatterns( in getCanonicalizationPatterns()