Searched defs:MatrixDiag (Results 1 – 4 of 4) sorted by relevance
81 struct MatrixDiag { struct85 generator::MatrixDiagGenerator<T> generator(input); in Compute() argument
145 struct MatrixDiag<CPUDevice, T> { struct146 static void Compute(const CPUDevice& d, in Compute()
2267 TEST_F(OpTest, MatrixDiag) { in TEST_F() argument
2585 func MatrixDiag(scope *Scope, diagonal tf.Output) (output tf.Output) { func