Searched refs:tf_dim_idx (Results 1 – 1 of 1) sorted by relevance
249 size_t tf_dim_idx(int index) const { return tf_to_mkl_dim_map_[index]; } in tf_dim_idx() function253 bool IsMklChannelDim(int d) const { return tf_dim_idx(d) == MklDims::C; } in IsMklChannelDim()256 bool IsMklBatchDim(int d) const { return tf_dim_idx(d) == MklDims::N; } in IsMklBatchDim()259 bool IsMklWidthDim(int d) const { return tf_dim_idx(d) == MklDims::W; } in IsMklWidthDim()262 bool IsMklHeightDim(int d) const { return tf_dim_idx(d) == MklDims::H; } in IsMklHeightDim()810 output_shape.AddDim(mkl_shape.GetSizes()[mkl_shape.tf_dim_idx(j)]); in ConvertMklToTF()