Home
last modified time | relevance | path

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

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