| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_softmax_op.cc | 57 // src_dims is the dimension of src_tensor
58 // dim of the dst will also be same as src_dims
63 auto src_dims = TFShapeToMklDnnDims(src_tf_shape); variable
71 output_dims = src_dims;
119 // (src_dims) required is in MKL-DNN order, the layout is Tensorflow's
123 : memory::desc(src_dims, MklDnnType<T>(), layout_type);
|
| mkl_concat_op.cc | 320 memory::dims src_dims(src_md.data.dims,
323 memory::desc(src_dims, MklDnnType<T>(), mkl_common_format);
333 memory::dims src_dims = TFShapeToMklDnnDims(input_tensors[k].shape()); local
334 dst_concat_dim_size += src_dims[concat_dim];
339 memory::desc(src_dims, MklDnnType<T>(), memory::format::nchw);
507 auto src_dims = TFShapeToMklDnnDims(input_shapes[k].GetTfShape()); local
508 *concat_dim_size += src_dims[concat_dim];
|
| mkl_avgpooling_op.cc | 483 memory::dims src_dims = variable
492 : memory::desc(src_dims, MklDnnType<T>(),
504 MklPoolingParams fwdParams(src_dims, output_dims_mkl_order, filter_dims,
523 {{src_dims}, MklDnnType<T>(), pooling_fwd->GetSrcMemoryFormat()},
|
| mkl_conv_grad_filter_ops.cc | 52 memory::dims src_dims; member in struct:tensorflow::MklConvBwdFilterParams
62 MklConvBwdFilterParams(memory::dims src_dims, memory::dims diff_filter_dims,
67 : src_dims(src_dims),
211 {convBwdFilterDims.src_dims}, MklDnnType<T>(), memory::format::any));
336 key_creator.AddAsKey(convBwdFilterDims.src_dims);
|
| mkl_maxpooling_op.cc | 561 memory::dims src_dims = variable
581 MklPoolingParams fwdParams(src_dims, output_dims_mkl_order, filter_dims,
600 {{src_dims}, MklDnnType<T>(), pooling_fwd->GetSrcMemoryFormat()},
[all...] |
| mkl_pooling_ops_common.h | 46 memory::dims src_dims; member in struct:tensorflow::MklPoolingParams
55 MklPoolingParams(memory::dims src_dims, memory::dims dst_dims,
59 : src_dims(src_dims),
188 key_creator.AddAsKey(fwdParams.src_dims);
340 key_creator.AddAsKey(bwdParams.src_dims);
|
| mkl_relu_op.cc | 43 memory::dims src_dims; // check if this is needed member in class:tensorflow::MklEltwiseFwdParams
49 MklEltwiseFwdParams(memory::dims src_dims, memory::desc src_md,
51 : src_dims(src_dims),
202 key_creator.AddAsKey(fwdParams.src_dims);
226 memory::dims src_dims; member in class:tensorflow::MklEltwiseBwdParams
232 MklEltwiseBwdParams(const memory::dims& src_dims,
235 : src_dims(src_dims),
421 key_creator.AddAsKey(bwdParams.src_dims);
471 memory::dims src_dims; variable
586 memory::dims src_dims = {}; local
[all...] |
| mkl_fused_batch_norm_op.cc | 36 memory::dims src_dims; member in struct:tensorflow::MklBatchNormFwdParams
41 MklBatchNormFwdParams(const memory::dims& src_dims, int depth, float eps,
43 : src_dims(src_dims), depth(depth), eps(eps), training(training) {}
147 auto src_md = memory::desc({fwdParams.src_dims}, MklDnnType<T>(),
148 get_desired_format(fwdParams.src_dims[1]));
252 key_creator.AddAsKey(fwdParams.src_dims);
272 memory::dims src_dims; member in struct:tensorflow::MklBatchNormBwdParams
278 MklBatchNormBwdParams(memory::dims src_dims, memory::dims diff_dst_dims,
280 : src_dims(src_dims)
610 memory::dims src_dims = variable
924 memory::dims src_dims = variable
[all...] |
| mkl_conv_ops.cc | 67 memory::dims src_dims; member in struct:tensorflow::MklConvFwdParams
82 MklConvFwdParams(memory::dims src_dims, memory::dims filter_dims,
86 : src_dims(src_dims),
217 {convFwdDims.src_dims}, MklDnnType<Tinput>(), memory::format::any));
365 key_creator.AddAsKey(convFwdDims.src_dims);
940 memory::dims src_dims, filter_dims, padding_left, padding_right, variable
[all...] |
