Home
last modified time | relevance | path

Searched refs:MklTensorFormat (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dmkl_types.h39 #define GET_FORMAT_FROM_SHAPE(src_mkl_shape) MklTensorFormat::FORMAT_BLOCKED
89 #define MKL_TENSOR_FORMAT MklTensorFormat
90 #define MKL_TENSOR_FORMAT_BLOCKED MklTensorFormat::FORMAT_BLOCKED
92 #define MKL_TENSOR_FORMAT_INVALID MklTensorFormat::FORMAT_INVALID
93 #define MKL_TENSOR_FORMAT_NC MklTensorFormat::FORMAT_NC
94 #define MKL_TENSOR_FORMAT_NCHW MklTensorFormat::FORMAT_NCHW
95 #define MKL_TENSOR_FORMAT_NCDHW MklTensorFormat::FORMAT_NCDHW
96 #define MKL_TENSOR_FORMAT_NDHWC MklTensorFormat::FORMAT_NDHWC
97 #define MKL_TENSOR_FORMAT_NHWC MklTensorFormat::FORMAT_NHWC
98 #define MKL_TENSOR_FORMAT_TNC MklTensorFormat::FORMAT_TNC
[all …]
Dmkl_util.h161 enum class MklTensorFormat { enum
174 memory::format_tag MklTensorFormatToMklDnnDataFormat(MklTensorFormat format);
176 TensorFormat MklDnn3DDataFormatToTFDataFormat(MklTensorFormat format);
177 TensorFormat MklDnnDataFormatToTFDataFormat(MklTensorFormat format);
197 inline void operator<<(std::ostream& os, const MklTensorFormat& format) {
198 if (format == MklTensorFormat::FORMAT_NHWC) {
200 } else if (format == MklTensorFormat::FORMAT_NCHW) {
202 } else if (format == MklTensorFormat::FORMAT_NDHWC) {
204 } else if (format == MklTensorFormat::FORMAT_NCDHW) {
206 } else if (format == MklTensorFormat::FORMAT_X) {
[all …]
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_softmax_op.cc39 MklTensorFormat src_fmt;
42 MklSoftmaxParams(memory::dims src_dims, MklTensorFormat src_fmt, int axis) in MklSoftmaxParams()
224 MklTensorFormat layout_type; in Compute()
236 layout_type = MklTensorFormat::FORMAT_X; in Compute()
239 layout_type = MklTensorFormat::FORMAT_NC; in Compute()
242 layout_type = MklTensorFormat::FORMAT_TNC; in Compute()
246 layout_type = MklTensorFormat::FORMAT_NHWC; in Compute()
248 layout_type = MklTensorFormat::FORMAT_NCHW; in Compute()
253 layout_type = MklTensorFormat::FORMAT_NDHWC; in Compute()
255 layout_type = MklTensorFormat::FORMAT_NCDHW; in Compute()
[all …]
Dmkl_conv_grad_input_ops.cc65 MklTensorFormat tf_fmt;
73 MklTensorFormat tf_fmt, bool native_format, in MklConvBwdInputParams()
553 MklTensorFormat GetOutputFormat(const MklTensorFormat data_format) { in GetOutputFormat()
562 MklTensorFormat output_tf_format, in AllocateOutputTensor()
Dmkl_reshape_op.cc57 MklTensorFormat::FORMAT_BLOCKED); in SkipReorder()
Dmkl_pooling_ops_common.h592 MklTensorFormat tensor_format_mkldnn_;
645 const MklTensorFormat& output_tf_format, in AllocateOutputTensor()
699 const MklTensorFormat& output_tf_format, in AllocateOutputTensor()
Dmkl_fused_batch_norm_op.cc752 MklTensorFormat mkl_tensor_fmt; in Compute()
756 mkl_tensor_fmt = MklTensorFormat::FORMAT_NCHW; in Compute()
759 mkl_tensor_fmt = MklTensorFormat::FORMAT_NHWC; in Compute()
1142 MklTensorFormat mkl_tensor_fmt; in Compute()
1146 mkl_tensor_fmt = MklTensorFormat::FORMAT_NCHW; in Compute()
1149 mkl_tensor_fmt = MklTensorFormat::FORMAT_NHWC; in Compute()
Dmkl_aggregate_ops.cc165 MklTensorFormat mkl_data_format; in Compute()
Dmkl_conv_ops.cc67 MklTensorFormat tf_fmt;
82 MklTensorFormat tf_fmt, bool native_format) in MklConvFwdParams()
851 MklTensorFormat output_tf_format, in AllocateOutputTensor()
966 MklTensorFormat GetFilterTfDataFormat(const MklDnnShape* filter_mkl_shape, in GetFilterTfDataFormat()
1027 MklTensorFormat::FORMAT_BLOCKED); in AllocateFilterOutputTensor()
1741 MklTensorFormat output_tf_format, in AllocateOutputTensor()
Dmkl_matmul_op_fused.cc163 MklTensorFormatToMklDnnDataFormat(MklTensorFormat::FORMAT_NC); in Compute()
Dmkl_lrn_op.cc193 const MklTensorFormat& output_tf_format, Tensor** output_tensor) { in AllocateOutputTensor()
456 const MklTensorFormat& output_tf_format, Tensor** output_tensor) { in AllocateOutputTensor()
Dmkl_conv_grad_filter_ops.cc61 MklTensorFormat tf_fmt;
70 MklTensorFormat tf_fmt, bool native_format, in MklConvBwdFilterParams()
Dmkl_relu_op.cc510 MklTensorFormat::FORMAT_BLOCKED); in Compute()
635 MklTensorFormat src_mkl_data_format = dnn_shape_src.GetTfDataFormat(); in Compute()
647 MklTensorFormat diff_dst_mkl_data_format = in Compute()
Dmkl_qmatmul_op.cc203 auto input_output_fmt_mkldnn = MklTensorFormat::FORMAT_NC; in Compute()
Dmkl_matmul_ops_common.h393 MklTensorFormat output_tf_format, Tensor** output_tensor) { in AllocateOutputTensor()