Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op_fused.cc78 const ContractionOutputMapper<T, Eigen::Index>& output_mapper, in operator ()()
130 std::function<void(const ContractionOutputMapper<T, Eigen::Index>&,
138 const ContractionOutputMapper<T, Eigen::Index>& output_mapper, in operator ()()
Dfused_eigen_output_kernels.h71 using ContractionOutputMapper = variable
209 const ContractionOutputMapper<Scalar, StorageIndex>& output_mapper, in operator()
236 const ContractionOutputMapper<Scalar, StorageIndex>& output_mapper,
269 const ContractionOutputMapper<Scalar, StorageIndex>& output_mapper,
312 const ContractionOutputMapper<Scalar, StorageIndex>& output_mapper,
Dconv_ops_fused_impl.h116 const ContractionOutputMapper<T, Eigen::Index>& output_mapper, in operator()
185 std::function<void(const ContractionOutputMapper<T, Eigen::Index>&,
193 const ContractionOutputMapper<T, Eigen::Index>& output_mapper, in operator()