Searched defs:MatrixSetDiag (Results 1 – 5 of 5) sorted by relevance
206 struct MatrixSetDiag<CPUDevice, T> { struct207 static void Compute(OpKernelContext* context, const CPUDevice& device, in Compute()
100 struct MatrixSetDiag<GPUDevice, Scalar> { struct101 static void Compute(OpKernelContext* context, const GPUDevice& device, in Compute()
27 struct MatrixSetDiag { struct
3322 TEST_F(OpTest, MatrixSetDiag) { in TEST_F() argument
24858 func MatrixSetDiag(scope *Scope, input tf.Output, diagonal tf.Output) (output tf.Output) { func