Home
last modified time | relevance | path

Searched defs:MatrixDiag (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dmatrix_diag_op_gpu.cu.cc73 struct MatrixDiag<GPUDevice, T> { struct
74 static void Compute(OpKernelContext* context, const GPUDevice& device, in Compute()
H A Dmatrix_diag_op.cc352 struct MatrixDiag<CPUDevice, T> { struct
353 static void Compute(OpKernelContext* context, const CPUDevice& device, in Compute()
H A Dmatrix_diag_op.h59 struct MatrixDiag { struct
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Drandomized_tests.cc3281 TEST_F(OpTest, MatrixDiag) { in TEST_F() argument
/aosp_15_r20/external/tensorflow/tensorflow/go/op/
H A Dwrappers.go24042 func MatrixDiag(scope *Scope, diagonal tf.Output) (output tf.Output) { func