| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_softmax_op.cc | 61 // src_dims is the dimenstion of src_tensor
62 // dim of the dst will also be same as src_dims
66 auto src_dims = TFShapeToMklDnnDims(src_tf_shape); variable
67 auto output_dims = src_dims;
76 // (src_dims) required is in MKL-DNN order, the layout is Tensorflow's
81 : memory::desc(src_dims, MklDnnType<T>(), memory::format::nc);
90 src.SetOpMemDesc(src_dims, memory::format::nc);
|
| mkl_relu_op.cc | 406 auto src_dims = TFShapeToMklDnnDims(src_tensor.shape()); variable
407 auto src_strides = CalculateTFStrides(src_dims);
409 src_md = MklDnnData<T>::CreateBlockedMemDesc(src_dims, src_strides);
513 auto src_dims = TFShapeToMklDnnDims(src_tensor.shape()); local
514 auto src_strides = CalculateTFStrides(src_dims);
515 src_md = MklDnnData<T>::CreateBlockedMemDesc(src_dims, src_strides);
541 auto src_dims = TFShapeToMklDnnDimsInNCHW(src_tensor.shape(), local
544 memory::desc(src_dims, MklDnnType<T>(), diff_dst_mkl_data_format);
[all...] |
| mkl_fused_batch_norm_op.cc | 804 memory::dims src_dims; variable
1134 memory::dims src_dims, diff_dst_dims; variable
[all...] |
| mkl_concat_op.cc | 689 memory::dims src_dims; local
691 // Same comment as dst_dims for src_dims.
692 src_dims = (is_mkl_tensor)
696 dst_concat_dim_size += src_dims[concat_dim];
702 memory::desc(src_dims, MklDnnType<T>(), memory::format::nchw);
[all...] |
| mkl_conv_ops.cc | 531 memory::dims src_dims, filter_dims, padding_l, padding_r, strides; variable
539 src_tf_shape, filter_tf_shape, &src_dims, &filter_dims, &strides,
572 // (src_dims) required is in MKL-DNN order, the layout is Tensorflow's
576 : memory::desc(src_dims, MklDnnType<T>(), tf_fmt);
593 src.SetOpMemDesc(src_dims, memory::format::any);
[all...] |
