Home
last modified time | relevance | path

Searched defs:ForEachIndex (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_util.h656 static void ForEachIndex(const Shape& shape, absl::Span<const int64_t> base, in ForEachIndex() function
681 static void ForEachIndex(const Shape& shape, const FnType& visitor_function) { in ForEachIndex() function
H A Dshape_util_test.cc580 TEST(ShapeUtilTest, ForEachIndex) { in TEST() argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dloops.cc82 StatusOr<std::vector<XlaOp>> ForEachIndex( in ForEachIndex() function