Searched refs:tf_to_mkl_dim_map_ (Results 1 – 1 of 1) sorted by relevance
163 if (tf_to_mkl_dim_map_) delete[] tf_to_mkl_dim_map_; in ~MklShape()206 if (tf_to_mkl_dim_map_ == nullptr) { in SetTfDimOrder()207 tf_to_mkl_dim_map_ = new size_t[dimension]; in SetTfDimOrder()210 tf_to_mkl_dim_map_[ii] = dimension - (ii + 1); in SetTfDimOrder()216 if (tf_to_mkl_dim_map_ == nullptr) { in SetTfDimOrder()217 tf_to_mkl_dim_map_ = new size_t[dimension]; in SetTfDimOrder()220 tf_to_mkl_dim_map_[ii] = tf_to_mkl_dim_map[ii]; in SetTfDimOrder()227 if (tf_to_mkl_dim_map_ == nullptr) { in SetTfDimOrder()228 tf_to_mkl_dim_map_ = new size_t[dimension]; in SetTfDimOrder()230 tf_to_mkl_dim_map_[GetTensorDimIndex<2>(data_format, 'W')] = MklDims::W; in SetTfDimOrder()[all …]