Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
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()
Dbatch_matmul_op_impl.h158 static ConstMatrixMap ConstTensorSliceToEigenMatrix(const Tensor& t,
180 auto x = ConstTensorSliceToEigenMatrix(in_x, x_batch_index);
181 auto y = ConstTensorSliceToEigenMatrix(in_y, y_batch_index);