Home
last modified time | relevance | path

Searched defs:head_outside_compiled_ops (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_extract_head_tail_outside_compilation.cc136 llvm::SmallSetVector<Operation*, 4> head_outside_compiled_ops; in FindOutsideCompiledOpsAtHead() local
182 llvm::ArrayRef<Operation*> head_outside_compiled_ops, in CreateHeadComputation()
205 llvm::SmallVector<Operation*, 4> head_outside_compiled_ops = in LiftHeadOutsideCompiledOps() local
Dtpu_host_computation_expansion.cc68 llvm::SmallSetVector<Operation*, 4> head_outside_compiled_ops; in ExpandHeadOutsideCompiledOps() local