Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_control_flow.cc190 Block* block, int offset, ArrayRef<Value> implicit_inputs, in ReplaceImplicitInputs() argument
193 implicit_input_elements.reserve(implicit_inputs.size()); in ReplaceImplicitInputs()
199 for (auto& implicit_input : llvm::enumerate(implicit_inputs)) { in ReplaceImplicitInputs()
219 llvm::SetVector<Value> implicit_inputs; in TupleImplicitInputs() local
220 getUsedValuesDefinedAbove(region, region, implicit_inputs); in TupleImplicitInputs()
221 llvm::ArrayRef<Value> implicit_inputs_ref = implicit_inputs.getArrayRef(); in TupleImplicitInputs()
319 llvm::SetVector<Value> implicit_inputs; in LowerWhileRegion() local
320 getUsedValuesDefinedAbove(op.getOperation()->getRegions(), implicit_inputs); in LowerWhileRegion()
321 inputs.append(implicit_inputs.begin(), implicit_inputs.end()); in LowerWhileRegion()
330 implicit_inputs.size()); in LowerWhileRegion()
[all …]