Searched refs:ForEachIndexWithStatus (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 629 static Status ForEachIndexWithStatus(const Shape& shape, in ForEachIndexWithStatus() function
645 static Status ForEachIndexWithStatus( in ForEachIndexWithStatus() function
648 return ShapeUtil::ForEachIndexWithStatus( in ForEachIndexWithStatus()
658 ForEachIndexWithStatus(shape, base, count, incr, in ForEachIndex()
669 static Status ForEachIndexWithStatus(const Shape& shape, in ForEachIndexWithStatus() function
673 return ForEachIndexWithStatus(shape, base, in ForEachIndexWithStatus()
680 ForEachIndexWithStatus(shape, [&](absl::Span<const int64> indices) { in ForEachIndex()
|
| D | shape_util_test.cc | 569 TEST(ShapeUtilTest, ForEachIndexWithStatus) { in TEST() argument
581 Status error_status = ShapeUtil::ForEachIndexWithStatus( in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1126 TF_RETURN_IF_ERROR(ShapeUtil::ForEachIndexWithStatus( in HandleGather()
1133 TF_RETURN_IF_ERROR(ShapeUtil::ForEachIndexWithStatus( in HandleGather()
1465 TF_RETURN_IF_ERROR(ShapeUtil::ForEachIndexWithStatus( in HandleSort()
|
| D | hlo_evaluator_typed_visitor.h | 1825 eval_status = ShapeUtil::ForEachIndexWithStatus( in HandleReduce()
2398 TF_RETURN_IF_ERROR(ShapeUtil::ForEachIndexWithStatus( in HandleScatter()
2407 TF_RETURN_IF_ERROR(ShapeUtil::ForEachIndexWithStatus( in HandleScatter()
|