HomeSort by relevance Sort by last modified time
    Searched refs:MklDims (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_conv_grad_bias_ops.cc 78 mkl_context.c_size = mkl_context.input_shape.GetSizes()[MklDims::C];
96 mkl_context.in_sizes[MklDims::W] =
97 mkl_context.input_shape.GetSizes()[MklDims::W];
98 mkl_context.in_sizes[MklDims::H] =
99 mkl_context.input_shape.GetSizes()[MklDims::H];
100 mkl_context.in_sizes[MklDims::C] =
101 mkl_context.input_shape.GetSizes()[MklDims::C];
102 mkl_context.in_sizes[MklDims::N] =
103 mkl_context.input_shape.GetSizes()[MklDims::N];
105 mkl_context.in_sizes[MklDims::W] = GetTensorDim(input, data_format_, 'W')
    [all...]
mkl_concat_op.cc 323 concat_dim = MklDims::C;
378 mkl_context.out_sizes[MklDims::W] = inpshape0.GetSizes()[MklDims::W];
379 mkl_context.out_sizes[MklDims::H] = inpshape0.GetSizes()[MklDims::H];
380 mkl_context.out_sizes[MklDims::C] = output_concat_dim_size;
381 mkl_context.out_sizes[MklDims::N] = inpshape0.GetSizes()[MklDims::N];
    [all...]
mkl_conv_ops.cc 213 mkl_context.in_sizes[MklDims::W] = static_cast<size_t>(input_cols);
214 mkl_context.in_sizes[MklDims::H] = static_cast<size_t>(input_rows);
215 mkl_context.in_sizes[MklDims::C] = static_cast<size_t>(input_depth);
216 mkl_context.in_sizes[MklDims::N] = static_cast<size_t>(batch);
218 mkl_context.out_sizes[MklDims::W] = static_cast<size_t>(out_cols);
219 mkl_context.out_sizes[MklDims::H] = static_cast<size_t>(out_rows);
220 mkl_context.out_sizes[MklDims::C] = static_cast<size_t>(out_depth);
221 mkl_context.out_sizes[MklDims::N] = static_cast<size_t>(batch);
    [all...]
mkl_lrn_op.cc 121 MklDims::C) {
397 if (tmp_mkl_shape->tf_dim_idx(mkl_context.in_dims - 1) != MklDims::C) {
    [all...]
  /external/tensorflow/tensorflow/core/util/
mkl_util.h 58 typedef enum { W = 0, H = 1, C = 2, N = 3 } MklDims;
145 tf_to_mkl_dim_map_[GetTensorDimIndex<2>(data_format, 'W')] = MklDims::W;
146 tf_to_mkl_dim_map_[GetTensorDimIndex<2>(data_format, 'H')] = MklDims::H;
147 tf_to_mkl_dim_map_[GetTensorDimIndex<2>(data_format, 'C')] = MklDims::C;
148 tf_to_mkl_dim_map_[GetTensorDimIndex<2>(data_format, 'N')] = MklDims::N;
168 bool IsMklChannelDim(int d) const { return tf_dim_idx(d) == MklDims::C; }
171 bool IsMklBatchDim(int d) const { return tf_dim_idx(d) == MklDims::N; }
174 bool IsMklWidthDim(int d) const { return tf_dim_idx(d) == MklDims::W; }
177 bool IsMklHeightDim(int d) const { return tf_dim_idx(d) == MklDims::H; }
    [all...]

Completed in 57 milliseconds