Searched refs:DimSize (Results 1 – 4 of 4) sorted by relevance
574 depth_ = dnn_shape_src.DimSize(MklDnnDims::Dim_C); in Compute()902 depth_ = dnn_shape_src.DimSize(MklDnnDims::Dim_C); in Compute()904 depth_ = dnn_shape_diff_dst.DimSize(MklDnnDims::Dim_C); in Compute()
686 original_input_mkl_shape.DimSize(0) == 4, in SanityCheckInputs()
516 return this->DimSize(index); in GetDimension()523 return this->DimSize(index); in GetDimension3D()581 inline int64 DimSize(int index) const { in DimSize() function
1099 template <std::ptrdiff_t DimSize = dynamic_range>1102 static const std::ptrdiff_t value = DimSize;