Home | History | Annotate | Download | only in util

Lines Matching defs:GetDimension

156   size_t GetDimension() const { return dimension_; }
426   inline size_t GetDimension(char dimension) const {
428     CHECK(index >= 0 && index < this->GetDimension())
449   inline size_t GetDimension() const { return data_.dimension_; }
684   for (size_t j = 0; j < mkl_shape.GetDimension(); j++) {
808   second_shape.AddDim(SIZE_OF_MKL_SERIAL_DATA(mkl_shape.GetDimension()));
842   second_shape.AddDim(SIZE_OF_MKL_SERIAL_DATA(mkl_shape.GetDimension()));
931   size_t tf_dim = mkl_shape.GetDimension();
971   CHECK(index >= 0 && index < mkl_shape.GetDimension())
1144   if (input_shape_0->GetDimension() != input_shape_1->GetDimension()) {
1149   size_t ndims = input_shape_0->GetDimension();
1164   if (input_shape_0->GetDimension() != input_shape_1->dims()) {
1169   size_t ndims = input_shape_0->GetDimension();