Searched refs:TfDimIdx (Results 1 – 2 of 2) sorted by relevance
594 shape[idx] = data_.sizes_[TfDimIdx(idx)]; in GetTfShape()706 inline size_t TfDimIdx(int index) const { return data_.map_[index]; } in TfDimIdx() function708 return data_.sizes_[TfDimIdx(index)]; in TfDimSize()714 return TfDimIdx(d) == MklDnnDims::Dim_C; in IsMklChannelDim()719 return TfDimIdx(d) == MklDnnDims::Dim_N; in IsMklBatchDim()724 return TfDimIdx(d) == MklDnnDims::Dim_W; in IsMklWidthDim()729 return TfDimIdx(d) == MklDnnDims::Dim_H; in IsMklHeightDim()
389 concat_dim = mkl_input_shapes[0].TfDimIdx(concat_dim); in Compute()