Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dregion_control_flow_to_functional.cc77 llvm::SetVector<Value> extern_values; in CollectExternValues() local
86 extern_values.insert(extern_value); in CollectExternValues()
97 return llvm::to_vector<4>(extern_values); in CollectExternValues()
107 llvm::SmallVectorImpl<Value>& extern_values, in ExtractSingleBlockRegion() argument
120 for (auto input : extern_values) input_types.push_back(input.getType()); in ExtractSingleBlockRegion()
126 for (auto input : extern_values) return_types.push_back(input.getType()); in ExtractSingleBlockRegion()
137 for (auto it : llvm::enumerate(extern_values)) { in ExtractSingleBlockRegion()
280 llvm::SmallVector<Value, 4> extern_values; in ConvertIfOp() local
288 extern_values.push_back(first); in ConvertIfOp()
304 extern_values = in ConvertIfOp()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dwhile_loop_outline.cc102 llvm::SetVector<Value> extern_values; in OutlineWhile() local
110 extern_values.insert(not_carried_operands.begin(), in OutlineWhile()
112 auto old_extern_values_size = extern_values.size(); in OutlineWhile()
122 extern_values.insert(extern_value); in OutlineWhile()
134 bool has_extra_extern_values = old_extern_values_size != extern_values.size(); in OutlineWhile()
141 extern_values.getArrayRef().drop_front(old_extern_values_size); in OutlineWhile()
152 for (Value operand : extern_values) types.push_back(operand.getType()); in OutlineWhile()
175 new_args.reserve(extern_values.size()); in OutlineWhile()
177 for (Value value : extern_values) { in OutlineWhile()
228 for (Type t : llvm::makeArrayRef(types).drop_back(extern_values.size())) in OutlineWhile()
[all …]