Searched refs:target_data_format (Results 1 – 4 of 4) sorted by relevance
137 StringRef target_data_format = force_data_format_; in runOnFunction() local138 if (target_data_format.empty()) { in runOnFunction()139 target_data_format = layout_sensitive_interface.GetOptimalLayout(devices); in runOnFunction()144 if (data_format == target_data_format) return; in runOnFunction()148 GetDataFormatPermutation(data_format, target_data_format); in runOnFunction()152 GetDataFormatPermutation(target_data_format, data_format); in runOnFunction()166 if (failed(layout_sensitive_interface.UpdateDataFormat(target_data_format))) in runOnFunction()
281 target_data_format='channels_first'): argument301 assert target_data_format in {'channels_last', 'channels_first'}304 if target_data_format == 'channels_first':
398 StringRef target_data_format = [&]() -> StringRef {407 if (target_data_format.empty()) return failure();411 // operation data format to `target_data_format`.421 GetDataFormatPermutation(op->data_format(), target_data_format);424 (*op)->setAttr("data_format", StringAttr::get(context, target_data_format));
903 target_data_format = ("NHWC" if data_format in (b"NCHW",912 "data_format": target_data_format,