Home
last modified time | relevance | path

Searched refs:rotate_amount (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter_transforms.cc140 int rotate_amount) { in SinkPerfectlyNestedLoops() argument
141 CHECK_GE(rotate_amount, 0); in SinkPerfectlyNestedLoops()
145 permutation.begin() + loops.size() - rotate_amount, in SinkPerfectlyNestedLoops()
Dconv_emitter_transforms.h97 int rotate_amount);