Home
last modified time | relevance | path

Searched defs:MatrixSetDiag (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_set_diag_op.h27 struct MatrixSetDiag { struct
Dmatrix_set_diag_op.cc203 struct MatrixSetDiag<CPUDevice, T> { struct
204 static void Compute(OpKernelContext* context, const CPUDevice& device, in Compute()
Dmatrix_set_diag_op_gpu.cu.cc100 struct MatrixSetDiag<GPUDevice, Scalar> { struct
101 static void Compute(OpKernelContext* context, const GPUDevice& device, in Compute()
/external/tensorflow/tensorflow/go/op/
Dwrappers.go10552 func MatrixSetDiag(scope *Scope, input tf.Output, diagonal tf.Output) (output tf.Output) { func