Home
last modified time | relevance | path

Searched refs:output_mapper (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfused_eigen_output_kernels.h209 const ContractionOutputMapper<Scalar, StorageIndex>& output_mapper, in operator()
218 T* output_base = &output_mapper(0, col); in operator()
236 const ContractionOutputMapper<Scalar, StorageIndex>& output_mapper,
245 T* output_base = &output_mapper(0, col);
269 const ContractionOutputMapper<Scalar, StorageIndex>& output_mapper,
284 T* output_base = &output_mapper(0, col);
312 const ContractionOutputMapper<Scalar, StorageIndex>& output_mapper,
327 T* output_base = &output_mapper(0, col);
Dmatmul_op_fused.cc78 const ContractionOutputMapper<T, Eigen::Index>& output_mapper, in operator ()()
81 output_kernel(output_mapper, params, i, j, num_rows, num_cols); in operator ()()
138 const ContractionOutputMapper<T, Eigen::Index>& output_mapper, in operator ()()
141 output_kernel_fn(output_mapper, params, i, j, num_rows, num_cols); in operator ()()
Deigen_contraction_kernel.h670 const OutputMapper& output_mapper, const LhsBlock& lhsBlock, \
677 GemmKernel()(output_mapper, lhsBlock.raw_data, rhsBlock.raw_data, \
684 GemmKernel()(output_mapper, lhsBlock.raw_data, rhsBlock.packed_data, \
691 GemmKernel()(output_mapper, lhsBlock.packed_data, rhsBlock.raw_data, \
698 GemmKernel()(output_mapper, lhsBlock.packed_data, \
709 ResScalar* output_base = &output_mapper(0, col); \
717 output_mapper, lhsBlock.packed_data, rhsBlock.packed_data, rows, \
854 const OutputMapper& output_mapper, const LhsBlock& lhsBlock, \
860 GemmKernel()(output_mapper, lhsBlock.raw_data, rhsBlock.raw_data, \
867 GemmKernel()(output_mapper, lhsBlock.raw_data, rhsBlock.packed_data, \
[all …]
Deigen_mkldnn_contraction_kernel_test.cc120 OutputMapper output_mapper(mkldnn_result.data(), m); in TEST() local
123 gemm_kernel(output_mapper, lhs.data(), rhs.data(), m, k, n, /*alpha=*/1.0, in TEST()
Dconv_ops_fused_impl.h113 const ContractionOutputMapper<T, Eigen::Index>& output_mapper, in operator()
116 output_kernel(output_mapper, params, i, j, num_rows, num_cols); in operator()
190 const ContractionOutputMapper<T, Eigen::Index>& output_mapper, in operator()
193 output_kernel_fn(output_mapper, params, i, j, num_rows, num_cols); in operator()