Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_ops.h141 mkldnn_sizes[MklDnnDims::Dim_N] = input_batch; in GetInputSizeInMklOrder()
142 mkldnn_sizes[MklDnnDims::Dim_C] = input_depth; in GetInputSizeInMklOrder()
143 mkldnn_sizes[MklDnnDims::Dim_H] = input_rows; in GetInputSizeInMklOrder()
144 mkldnn_sizes[MklDnnDims::Dim_W] = input_cols; in GetInputSizeInMklOrder()
238 mkldnn_sizes[MklDnnDims::Dim_O] = filter_out_depth; in GetFilterSizeInMklOrder()
239 mkldnn_sizes[MklDnnDims::Dim_I] = filter_in_depth; in GetFilterSizeInMklOrder()
240 mkldnn_sizes[MklDnnDims::Dim_H] = filter_rows; in GetFilterSizeInMklOrder()
241 mkldnn_sizes[MklDnnDims::Dim_W] = filter_cols; in GetFilterSizeInMklOrder()
456 mkldnn_sizes[MklDnnDims::Dim_N] = out_batch;
457 mkldnn_sizes[MklDnnDims::Dim_C] = out_depth;
[all …]
Dmkl_conv_grad_filter_ops.cc491 {bwd_output_dims[MklDnnDims::Dim_H], in Compute()
492 bwd_output_dims[MklDnnDims::Dim_W], in Compute()
493 bwd_output_dims[MklDnnDims::Dim_I], in Compute()
494 bwd_output_dims[MklDnnDims::Dim_O]}); in Compute()
681 TensorShape output_tf_shape({output_dims_mkl_order[MklDnnDims::Dim_H], in AllocateOutputTensor()
682 output_dims_mkl_order[MklDnnDims::Dim_W], in AllocateOutputTensor()
683 output_dims_mkl_order[MklDnnDims::Dim_I], in AllocateOutputTensor()
684 output_dims_mkl_order[MklDnnDims::Dim_O]}); in AllocateOutputTensor()
Dmkl_fused_batch_norm_op.cc574 depth_ = dnn_shape_src.DimSize(MklDnnDims::Dim_C); in Compute()
902 depth_ = dnn_shape_src.DimSize(MklDnnDims::Dim_C); in Compute()
904 depth_ = dnn_shape_diff_dst.DimSize(MklDnnDims::Dim_C); in Compute()
Dmkl_conv_ops.cc1061 (src_dims[MklDnnDims::Dim_N] > kSmallBatchSize) && in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h99 } MklDnnDims; typedef
529 return MklDnnDims::Dim_N; in GetMklDnnTensorDimIndex()
531 return MklDnnDims::Dim_C; in GetMklDnnTensorDimIndex()
533 return MklDnnDims::Dim_H; in GetMklDnnTensorDimIndex()
535 return MklDnnDims::Dim_W; in GetMklDnnTensorDimIndex()
693 data_.map_[GetTensorDimIndex<2>(data_format, 'W')] = MklDnnDims::Dim_W; in SetTfDimOrder()
694 data_.map_[GetTensorDimIndex<2>(data_format, 'H')] = MklDnnDims::Dim_H; in SetTfDimOrder()
695 data_.map_[GetTensorDimIndex<2>(data_format, 'C')] = MklDnnDims::Dim_C; in SetTfDimOrder()
696 data_.map_[GetTensorDimIndex<2>(data_format, 'N')] = MklDnnDims::Dim_N; in SetTfDimOrder()
714 return TfDimIdx(d) == MklDnnDims::Dim_C; in IsMklChannelDim()
[all …]