Home
last modified time | relevance | path

Searched refs:num_dims_i (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/c/experimental/gradients/tape/
Dtape_operation.cc168 const auto num_dims_i = num_dims[i]; in SetAttrShapeList() local
170 if (num_dims_i > TensorShape::MaxDimensions()) { in SetAttrShapeList()
173 num_dims_i, " dimensions which is over the limit of ", in SetAttrShapeList()
176 if (num_dims_i < 0) { in SetAttrShapeList()
181 for (int d = 0; d < num_dims_i; ++d) { in SetAttrShapeList()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_operation.cc177 const auto num_dims_i = num_dims[i]; in SetAttrShapeList() local
179 if (num_dims_i > TensorShape::MaxDimensions()) { in SetAttrShapeList()
182 num_dims_i, " dimensions which is over the limit of ", in SetAttrShapeList()
185 if (num_dims_i < 0) { in SetAttrShapeList()
190 for (int d = 0; d < num_dims_i; ++d) { in SetAttrShapeList()
/external/tensorflow/tensorflow/c/eager/
Dgradients.cc437 const auto num_dims_i = num_dims[i]; in SetAttrShapeList() local
439 if (num_dims_i > TensorShape::MaxDimensions()) { in SetAttrShapeList()
442 num_dims_i, " dimensions which is over the limit of ", in SetAttrShapeList()
445 if (num_dims_i < 0) { in SetAttrShapeList()
450 for (int d = 0; d < num_dims_i; ++d) { in SetAttrShapeList()