Searched refs:ForEachIndexParallel (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 694 static void ForEachIndexParallel(const Shape& shape, in ForEachIndexParallel() function
713 static void ForEachIndexParallel(const Shape& shape, in ForEachIndexParallel() function
717 return ForEachIndexParallel(shape, base, in ForEachIndexParallel()
|
| D | shape_util_test.cc | 630 TEST(ShapeUtilTest, ForEachIndexParallel) { in TEST() argument
638 ShapeUtil::ForEachIndexParallel(shape, /*base=*/{0, 0}, /*count=*/{10, 10}, in TEST()
|
| D | literal.h | 1405 ShapeUtil::ForEachIndexParallel(this_shape, stride_config.base, in PopulateInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_typed_visitor.h | 1870 ShapeUtil::ForEachIndexParallel(
|