MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / AddLoopsForShape

Method AddLoopsForShape

tensorflow/compiler/xla/service/llvm_ir/llvm_loop.cc:234–240  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

232}
233
234IrArray::Index ForLoopNest::AddLoopsForShape(const Shape& shape,
235 absl::string_view suffix) {
236 std::vector<int64> dimensions(shape.rank());
237 std::iota(dimensions.begin(), dimensions.end(), 0);
238 return IrArray::Index(AddLoopsForShapeOnDimensions(shape, dimensions, suffix),
239 shape, index_type_);
240}
241
242std::vector<llvm::Value*> ForLoopNest::AddLoopsForShapeOnDimensions(
243 const Shape& shape, absl::Span<const int64> dimensions,

Callers 5

MakeElementGeneratorMethod · 0.80
HandlePadMethod · 0.80

Calls 4

IndexClass · 0.70
rankMethod · 0.45
beginMethod · 0.45
endMethod · 0.45

Tested by

no test coverage detected