| /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 | 114 using ConstMatrixMap = Eigen::Map<const Matrix>; variable
116 using ConstMatrixMaps = gtl::InlinedVector<ConstMatrixMap, 4>;
181 using ConstMatrixMap = typename Base::ConstMatrixMap; \
|
| D | matrix_triangular_solve_op.cc | 82 const ConstMatrixMap& matrix = inputs[0]; in ComputeMatrix()
83 const ConstMatrixMap& rhs = inputs[1]; in ComputeMatrix()
174 const ConstMatrixMap& matrix = inputs[0]; in ComputeMatrix()
175 const ConstMatrixMap& rhs = inputs[1]; in ComputeMatrix()
|
| 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 | 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 | lu_op.cc | 40 using ConstMatrixMap = Eigen::Map<const Matrix>; typedef in tensorflow::LuOp
134 ConstMatrixMap input_matrix( in ComputeTensorSlice()
|
| D | matrix_logarithm_op.cc | 40 const ConstMatrixMap& input = inputs[0]; in ComputeMatrix()
|
| D | matrix_exponential_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 | 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 | batch_matmul_op_impl.h | 156 using ConstMatrixMap = Eigen::Map<const Matrix>;
159 static ConstMatrixMap ConstTensorSliceToEigenMatrix(const Tensor& t,
161 return ConstMatrixMap(
|
| D | matrix_inverse_op.cc | 53 const ConstMatrixMap& input = inputs[0]; in ComputeMatrix()
|
| D | conv_grad_input_ops.cc | 194 using ConstMatrixMap = Eigen::Map< typedef
203 ConstMatrixMap A(out_data, output_image_size, dims_out_depth); in operator ()()
204 ConstMatrixMap B(filter_data, filter_total_size, dims_out_depth); in operator ()()
|
| D | conv_grad_ops_3d.cc | 517 ConstMatrixMap; in Compute() typedef
539 ConstMatrixMap A(out_data, output_image_size, dims.out_depth); in Compute()
540 ConstMatrixMap B(filter_data, filter_total_size, dims.out_depth); in Compute()
|
| D | sparse_matmul_op.cc | 113 using ConstMatrixMap = BasicMatrixMap<const T>; typedef
141 void Initialize(const ConstMatrixMap& mat, int col_offset);
180 const typename SparseSlice<T>::ConstMatrixMap& mat, int col_offset) { in Initialize()
|
| /external/tensorflow/tensorflow/contrib/nearest_neighbor/kernels/ |
| D | hyperplane_lsh_probes.cc | 37 using ConstMatrixMap = Eigen::Map<const Matrix>; typedef in tensorflow::HyperplaneLSHProbesOp
101 ConstMatrixMap products_matrix(products_eigen_tensor.data(), in Compute()
|
| D | hyperplane_lsh_probes.h | 40 using ConstMatrixMap = Eigen::Map<const Matrix>; variable
|