Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dlayout_util.cc23 const LayoutPreferenceFn& layout_preference_fn, in RewriteLayoutWithShardedShape()
68 const LayoutPreferenceFn& layout_preference_fn, in ReshapeWithCorrectRepresentationAndSharding()
Dmlir_hlo_to_hlo.h47 LayoutPreferenceFn layout_preference_fn; member
/external/tensorflow/tensorflow/compiler/tf2xla/
Dlayout_util.h54 LayoutPreferenceFn layout_preference_fn; member