Searched refs:tf_data_format (Results 1 – 2 of 2) sorted by relevance
2956 tf_data_format = 'NHWC'2959 tf_data_format = 'NCHW'2969 x, gamma, beta, epsilon=epsilon, data_format=tf_data_format)3028 tf_data_format = 'NCHW'3030 tf_data_format = 'NHWC'3032 tf_data_format = None3034 if (tf_data_format == 'NHWC' or3035 tf_data_format == 'NCHW' and _has_nchw_support()):3057 data_format=tf_data_format,5252 tf_data_format = 'NWC' # to pass TF Conv2dNative operations[all …]
166 TensorFormat tf_data_format; in Compute() local173 tf_data_format = MklDnnDataFormatToTFDataFormat(mkl_data_format); in Compute()197 TFShapeToMklDnnDimsInNCHW(src_tensor.shape(), tf_data_format); in Compute()201 tf_data_format); in Compute()