Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/dtensor/mlir/
Ddtensor_allreduce_combine_optimization.cc193 std::vector<mlir::Type> flattened_types; in MergeAllReduceGroup() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc193 Type type, llvm::SmallVectorImpl<Type>& flattened_types) { in FlattenTupleType()
807 llvm::SmallVector<Type> flattened_types; in ImportInstructionImpl() local