Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_dynamic_layout_pass.cc124 TF::TPUGetLayoutOp BuildGetLayout(const int64_t execute_arg_index, in BuildGetLayout() function
154 auto get_layout = BuildGetLayout(execute_arg_index, execute.key(), in HandleInput()
181 auto get_layout = BuildGetLayout(execute_arg_index, compilation_key, in HandleReplicatedInputs()