HomeSort by relevance Sort by last modified time
    Searched defs:MatrixDiagPart (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
matrix_diag_op.h 71 struct MatrixDiagPart {
matrix_diag_op.cc 75 functor::MatrixDiagPart<Device, T>::Compute(
124 Name("MatrixDiagPart").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
159 struct MatrixDiagPart<CPUDevice, T> {
184 void MatrixDiagPart<GPUDevice, T>::Compute( \
187 extern template struct MatrixDiagPart<GPUDevice, T>;
202 Name("MatrixDiagPart").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 102 milliseconds