OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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