Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_diag_op.h48 struct MatrixDiagPart { struct
Dmatrix_diag_op_gpu.cu.cc134 struct MatrixDiagPart<GPUDevice, T> { struct
135 static void Compute(OpKernelContext* context, const GPUDevice& device, in Compute()
Dmatrix_diag_op.cc386 struct MatrixDiagPart<CPUDevice, T> { struct
387 static void Compute(OpKernelContext* context, const CPUDevice& device, in Compute()
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc2277 TEST_F(OpTest, MatrixDiagPart) { in TEST_F() argument
/external/tensorflow/tensorflow/go/op/
Dwrappers.go2388 func MatrixDiagPart(scope *Scope, input tf.Output) (diagonal tf.Output) { func