Searched refs:tf_data_format_ (Results 1 – 1 of 1) sorted by relevance
255 MklTensorFormat tf_data_format_ = MklTensorFormat::FORMAT_BLOCKED; member403 if (data_.tf_data_format_ != MklTensorFormat::FORMAT_BLOCKED) { in GetTfShape()435 return data_.tf_data_format_; in GetTfDataFormat()450 data_.tf_data_format_ = format; in SetTfLayout()468 if (data_.tf_data_format_ == MklTensorFormat::FORMAT_BLOCKED) { in GetTfLayout()476 MklTensorFormatToMklDnnDataFormat(data_.tf_data_format_); in GetTfLayout()