Searched refs:MklDims (Results 1 – 3 of 3) sorted by relevance
81 mkl_context.c_size = mkl_context.input_shape.GetSizes()[MklDims::C]; in Compute()100 mkl_context.in_sizes[MklDims::W] = in Compute()101 mkl_context.input_shape.GetSizes()[MklDims::W]; in Compute()102 mkl_context.in_sizes[MklDims::H] = in Compute()103 mkl_context.input_shape.GetSizes()[MklDims::H]; in Compute()104 mkl_context.in_sizes[MklDims::C] = in Compute()105 mkl_context.input_shape.GetSizes()[MklDims::C]; in Compute()106 mkl_context.in_sizes[MklDims::N] = in Compute()107 mkl_context.input_shape.GetSizes()[MklDims::N]; in Compute()109 mkl_context.in_sizes[MklDims::W] = GetTensorDim(input, data_format_, 'W'); in Compute()[all …]
564 mkl_context.in_sizes[MklDims::W] = static_cast<size_t>(input_cols); in Compute()565 mkl_context.in_sizes[MklDims::H] = static_cast<size_t>(input_rows); in Compute()566 mkl_context.in_sizes[MklDims::C] = static_cast<size_t>(input_depth); in Compute()567 mkl_context.in_sizes[MklDims::N] = static_cast<size_t>(batch); in Compute()569 mkl_context.out_sizes[MklDims::W] = static_cast<size_t>(out_cols); in Compute()570 mkl_context.out_sizes[MklDims::H] = static_cast<size_t>(out_rows); in Compute()571 mkl_context.out_sizes[MklDims::C] = static_cast<size_t>(out_depth); in Compute()572 mkl_context.out_sizes[MklDims::N] = static_cast<size_t>(batch); in Compute()
87 typedef enum { W = 0, H = 1, C = 2, N = 3 } MklDims; typedef230 tf_to_mkl_dim_map_[GetTensorDimIndex<2>(data_format, 'W')] = MklDims::W; in SetTfDimOrder()231 tf_to_mkl_dim_map_[GetTensorDimIndex<2>(data_format, 'H')] = MklDims::H; in SetTfDimOrder()232 tf_to_mkl_dim_map_[GetTensorDimIndex<2>(data_format, 'C')] = MklDims::C; in SetTfDimOrder()233 tf_to_mkl_dim_map_[GetTensorDimIndex<2>(data_format, 'N')] = MklDims::N; in SetTfDimOrder()253 bool IsMklChannelDim(int d) const { return tf_dim_idx(d) == MklDims::C; } in IsMklChannelDim()256 bool IsMklBatchDim(int d) const { return tf_dim_idx(d) == MklDims::N; } in IsMklBatchDim()259 bool IsMklWidthDim(int d) const { return tf_dim_idx(d) == MklDims::W; } in IsMklWidthDim()262 bool IsMklHeightDim(int d) const { return tf_dim_idx(d) == MklDims::H; } in IsMklHeightDim()1118 return MklDims::N; in GetMklTensorDimIndex()[all …]