HomeSort by relevance Sort by last modified time
    Searched refs:MatrixBandPartOp (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
matrix_band_part_op.cc 26 class MatrixBandPartOp : public XlaOpKernel {
28 explicit MatrixBandPartOp(OpKernelConstruction* context)
92 TF_DISALLOW_COPY_AND_ASSIGN(MatrixBandPartOp);
94 REGISTER_XLA_OP(Name("MatrixBandPart"), MatrixBandPartOp);
  /external/tensorflow/tensorflow/core/kernels/
matrix_band_part_op.cc 46 class MatrixBandPartOp : public OpKernel {
48 explicit MatrixBandPartOp(OpKernelConstruction* context)
109 TF_DISALLOW_COPY_AND_ASSIGN(MatrixBandPartOp);
115 MatrixBandPartOp<CPUDevice, type>);
125 MatrixBandPartOp<CPUDevice, type>);
226 MatrixBandPartOp<GPUDevice, type>);
241 MatrixBandPartOp<GPUDevice, type>);

Completed in 125 milliseconds