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

  /external/tensorflow/tensorflow/core/kernels/
matrix_diag_op_gpu.cu.cc 28 template class generator::MatrixDiagGenerator<T>; \
matrix_diag_op.h 48 class MatrixDiagGenerator {
51 MatrixDiagGenerator(typename TTypes<T, 2>::ConstTensor input)
85 generator::MatrixDiagGenerator<T> generator(input);

Completed in 134 milliseconds