| /external/tensorflow/tensorflow/core/kernels/ |
| D | matrix_solve_ls_op_impl.h | 46 using ConstMatrixMap = typename Base::ConstMatrixMap; variable
78 const ConstMatrixMap& matrix = inputs[0]; in ComputeMatrix()
79 const ConstMatrixMap& rhs = inputs[1]; in ComputeMatrix()
|
| D | cholesky_grad.cc | 36 using ConstMatrixMap = typename Base::ConstMatrixMap; typedef in tensorflow::CholeskyGrad
62 const ConstMatrixMap& input_matrix_l_full = inputs[0]; in ComputeMatrix()
63 const ConstMatrixMap& input_matrix_grad = inputs[1]; in ComputeMatrix()
|
| D | linalg_ops_common.h | 136 using ConstMatrixMap = Eigen::Map<const Matrix>; variable
140 using ConstMatrixMaps = gtl::InlinedVector<ConstMatrixMap, 4>;
205 using ConstMatrixMap = typename Base::ConstMatrixMap; \
|
| D | self_adjoint_eig_op.cc | 41 using ConstMatrixMap = typename Base::ConstMatrixMap; typedef in tensorflow::SelfAdjointEigOp
|
| D | self_adjoint_eig_v2_op_impl.h | 46 using ConstMatrixMap = typename Base::ConstMatrixMap; variable
|
| D | svd_op_impl.h | 81 using ConstMatrixMap = typename Base::ConstMatrixMap; variable
|
| D | tridiagonal_matmul_op.cc | 106 ConstMatrixMap rhs_except_first_row(rhs.data() + n, m - 1, n); in ComputeMatrix()
107 ConstMatrixMap rhs_except_last_row(rhs.data(), m - 1, n); in ComputeMatrix()
|
| D | matrix_triangular_solve_op_impl.h | 67 using ConstMatrixMap = Eigen::Map<const Matrix>; member
71 static ConstMatrixMap ConstTensorSliceToEigenMatrix(const Tensor& t, in ConstTensorSliceToEigenMatrix()
73 return ConstMatrixMap( in ConstTensorSliceToEigenMatrix()
131 using ConstMatrixMap = Eigen::Map<const Matrix>;
134 auto matrix = ConstMatrixMap(in_x.flat<Scalar>().data(), in_x.dim_size(1),
|
| D | lu_op.cc | 40 using ConstMatrixMap = Eigen::Map<const Matrix>; typedef in tensorflow::LuOp
134 ConstMatrixMap input_matrix( in ComputeTensorSlice()
|
| D | deep_conv2d.cc | 195 ConstMatrixMap; typedef
214 ConstMatrixMap A(transform_matrix, tile_spatial_size, in operator ()()
216 ConstMatrixMap B(filter_in, base_filter_spatial_size, in_stride); in operator ()()
691 ConstMatrixMap; typedef
715 ConstMatrixMap A(transform_matrix, tile_spatial_size, tile_spatial_size); in operator ()()
716 ConstMatrixMap B(tile_buffer, tile_spatial_size, coord_stride); in operator ()()
743 ConstMatrixMap; typedef
763 ConstMatrixMap A(out_transform_matrix, out_tile_spatial_size, in operator ()()
765 ConstMatrixMap B(out_buffer, tile_spatial_size, out_buf_stride); in operator ()()
|
| D | matrix_logarithm_op.cc | 40 const ConstMatrixMap& input = inputs[0]; in ComputeMatrix()
|
| D | matrix_square_root_op.cc | 40 const ConstMatrixMap& input = inputs[0]; in ComputeMatrix()
|
| D | matrix_exponential_op.cc | 40 const ConstMatrixMap& input = inputs[0]; in ComputeMatrix()
|
| D | tridiagonal_solve_op.cc | 142 const ConstMatrixMap& rhs, in SolveWithGaussianEliminationWithPivoting()
198 const ConstMatrixMap& rhs, MatrixMap& x) { in SolveWithThomasAlgorithm()
|
| D | qr_op_impl.h | 98 using ConstMatrixMap = typename Base::ConstMatrixMap; variable
|
| D | matrix_solve_op.cc | 77 const ConstMatrixMap& matrix = inputs[0]; in ComputeMatrix()
78 const ConstMatrixMap& rhs = inputs[1]; in ComputeMatrix()
|
| D | cholesky_op.cc | 54 const ConstMatrixMap& input = inputs[0]; in ComputeMatrix()
|
| D | matrix_inverse_op.cc | 53 const ConstMatrixMap& input = inputs[0]; in ComputeMatrix()
|
| D | batch_matmul_op_impl.h | 155 using ConstMatrixMap = Eigen::Map<const Matrix>;
158 static ConstMatrixMap ConstTensorSliceToEigenMatrix(const Tensor& t,
160 return ConstMatrixMap(
|
| D | conv_grad_input_ops.cc | 311 using ConstMatrixMap = Eigen::Map< typedef
320 ConstMatrixMap A(out_data, output_image_size, dims_out_depth); in operator ()()
321 ConstMatrixMap B(filter_data, filter_total_size, dims_out_depth); in operator ()()
|
| D | sparse_matmul_op.cc | 112 using ConstMatrixMap = BasicMatrixMap<const T>; typedef
140 void Initialize(const ConstMatrixMap& mat, int col_offset);
179 const typename SparseSlice<T>::ConstMatrixMap& mat, int col_offset) { in Initialize()
|
| D | conv_grad_ops_3d.cc | 527 ConstMatrixMap; in Compute() typedef
549 ConstMatrixMap A(out_data, output_image_size, dims.out_depth); in Compute()
550 ConstMatrixMap B(filter_data, filter_total_size, dims.out_depth); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 160 using ConstMatrixMap = Eigen::Map<const Matrix>; typedef in tensorflow::CSRMatMulCPUOp
317 ConstMatrixMap rhs_map(rhs.flat<T>().data() + batch_idx * in SparseDenseMatMulWithoutTransposedLHS()
389 ConstMatrixMap b_dense_map( in SparseDenseMatMulWithTransposedLHS()
490 using ConstMatrixMap = Eigen::Map<const Matrix>; typedef in tensorflow::CSRMatMulGPUOp
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 31 using ConstMatrixMap = Eigen::Map<const Matrix>; typedef
307 ConstMatrixMap stats_mat(&stats_summary(node_id, 0, 0, 0), in Compute()
664 ConstMatrixMap stats_mat(&stats_summaries[0](node_id, 0, 0, 0), in Compute()
|