Home
last modified time | relevance | path

Searched refs:ConstTensorSliceToEigenMatrix (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dbanded_triangular_solve_op.cc49 static ConstMatrixMap ConstTensorSliceToEigenMatrix(const Tensor& t, in ConstTensorSliceToEigenMatrix() function
74 auto matrix = ConstTensorSliceToEigenMatrix(in_x, x_batch_index); in Run()
75 auto rhs = ConstTensorSliceToEigenMatrix(in_y, y_batch_index); in Run()
Dmatrix_triangular_solve_op_impl.h71 static ConstMatrixMap ConstTensorSliceToEigenMatrix(const Tensor& t, in ConstTensorSliceToEigenMatrix() function
93 auto matrix = ConstTensorSliceToEigenMatrix(in_x, x_batch_index); in Run()
94 auto rhs = ConstTensorSliceToEigenMatrix(in_y, y_batch_index); in Run()
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op_impl.h165 static ConstMatrixMap ConstTensorSliceToEigenMatrix(const Tensor& t,
187 auto x = ConstTensorSliceToEigenMatrix(in_x, x_batch_index);
188 auto y = ConstTensorSliceToEigenMatrix(in_y, y_batch_index);