Searched refs:ForEachIndexWithStatus (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 713 static Status ForEachIndexWithStatus(const Shape& shape, in ForEachIndexWithStatus() function
729 static Status ForEachIndexWithStatus( in ForEachIndexWithStatus() function
732 return ShapeUtil::ForEachIndexWithStatus( in ForEachIndexWithStatus()
742 ForEachIndexWithStatus(shape, base, count, incr, in ForEachIndex()
753 static Status ForEachIndexWithStatus(const Shape& shape, in ForEachIndexWithStatus() function
757 return ForEachIndexWithStatus(shape, base, in ForEachIndexWithStatus()
764 ForEachIndexWithStatus(shape, [&](absl::Span<const int64> indices) { in ForEachIndex()
|
| D | shape_util_test.cc | 593 TEST(ShapeUtilTest, ForEachIndexWithStatus) { in TEST() argument
605 Status error_status = ShapeUtil::ForEachIndexWithStatus( in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1800 TF_RETURN_IF_ERROR(ShapeUtil::ForEachIndexWithStatus( in HandleGather()
1807 TF_RETURN_IF_ERROR(ShapeUtil::ForEachIndexWithStatus( in HandleGather()
2185 TF_RETURN_IF_ERROR(ShapeUtil::ForEachIndexWithStatus( in HandleSort()
2386 TF_RETURN_IF_ERROR(ShapeUtil::ForEachIndexWithStatus( in GenerateReduceOutputElement()
2394 TF_RETURN_IF_ERROR(ShapeUtil::ForEachIndexWithStatus( in GenerateReduceOutputElement()
2473 TF_RETURN_IF_ERROR(ShapeUtil::ForEachIndexWithStatus( in HandleReduce()
|
| D | hlo_evaluator_typed_visitor.h | 2037 TF_RETURN_IF_ERROR(ShapeUtil::ForEachIndexWithStatus(
2419 TF_RETURN_IF_ERROR(ShapeUtil::ForEachIndexWithStatus(
2428 TF_RETURN_IF_ERROR(ShapeUtil::ForEachIndexWithStatus(
|