Home
last modified time | relevance | path

Searched refs:MaxDimensions (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/c/experimental/gradients/tape/
Dtape_operation.cc89 if (num_dims > TensorShape::MaxDimensions()) { in SetAttrShape()
93 TensorShape::MaxDimensions(), "."); in SetAttrShape()
170 if (num_dims_i > TensorShape::MaxDimensions()) { in SetAttrShapeList()
174 TensorShape::MaxDimensions(), ".")); in SetAttrShapeList()
/external/tensorflow/tensorflow/core/framework/
Dtensor_shape.cc64 if (dims() > MaxDimensions()) return false; in IsValid()
84 if (proto.dim().size() > MaxDimensions()) return false; in IsValid()
110 if (proto.dim().size() > MaxDimensions()) { in IsValidShape()
399 CHECK_LT(ndims_byte(), MaxDimensions()) << "Too many dimensions in tensor"; in AddDim()
422 if (TF_PREDICT_FALSE(ndims_byte() >= MaxDimensions())) { in AddDimWithStatus()
510 CHECK_LT(dims(), MaxDimensions()); in InsertDim()
536 if (TF_PREDICT_FALSE(dims() >= MaxDimensions())) { in InsertDimWithStatus()
787 if (n > TensorShape::MaxDimensions()) { in MakeShapeHelper()
Dtensor_shape.h64 static constexpr int MaxDimensions() { return 254; } in MaxDimensions() function
592 static_assert(NDIMS <= TensorShape::MaxDimensions(), "Too many dimensions"); in AsEigenDSizesCopyAndPad()
Dtensor_shape_test.cc145 for (int i = 0; i < TensorShape::MaxDimensions(); i++) { in TEST()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_operation.cc77 if (num_dims > TensorShape::MaxDimensions()) { in SetAttrShape()
81 TensorShape::MaxDimensions(), "."); in SetAttrShape()
179 if (num_dims_i > TensorShape::MaxDimensions()) { in SetAttrShapeList()
183 TensorShape::MaxDimensions(), ".")); in SetAttrShapeList()
/external/tensorflow/tensorflow/c/eager/
Dgradients.cc352 if (num_dims > TensorShape::MaxDimensions()) { in SetAttrShape()
356 TensorShape::MaxDimensions(), "."); in SetAttrShape()
439 if (num_dims_i > TensorShape::MaxDimensions()) { in SetAttrShapeList()
443 TensorShape::MaxDimensions(), ".")); in SetAttrShapeList()