Searched defs:MatrixDiagPart (Results 1 – 4 of 4) sorted by relevance
71 struct MatrixDiagPart { struct75 generator::MatrixDiagPartGenerator<T> generator(input); in Compute() argument
159 struct MatrixDiagPart<CPUDevice, T> { struct160 static void Compute(const CPUDevice& d, in Compute()
2276 TEST_F(OpTest, MatrixDiagPart) { in TEST_F() argument
2538 func MatrixDiagPart(scope *Scope, input tf.Output) (diagonal tf.Output) { func