Home
last modified time | relevance | path

Searched refs:propagate_layouts (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.h37 bool propagate_layouts = false; member
Dxla_mlir_translate.cc177 options.propagate_layouts = with_layouts; in MlirHloToHloTextTranslateFunctionImpl()
Dmlir_hlo_to_hlo.cc1217 auto propagate_layouts = [this](mlir::Operation* inst, in Lower() local
1219 if (options_.propagate_layouts) { in Lower()
1254 if (failed(propagate_layouts(inst, iter->second))) { in Lower()
1283 if (failed(propagate_layouts(inst, xla_operand))) { in Lower()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2749 options.propagate_layouts = true; in GetOrCreateSubComputationFromRegion()