Searched refs:eigenvalues_shape (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dself_adjoint_eig_v2_op_gpu.cc66 TensorShape eigenvalues_shape = input.shape(); variable
67 eigenvalues_shape.RemoveLastDims(1);
69 context, context->allocate_output(0, eigenvalues_shape, &eigenvalues),
94 eigenvalues_shape, &eigenvalues_real),

Completed in 267 milliseconds