Searched refs:function_block (Results 1 – 2 of 2) sorted by relevance
114 Block& function_block = func.front(); in VerifyShardings() local116 llvm::zip(sharding_for_args, function_block.getArguments())) { in VerifyShardings()121 Operation* terminator = function_block.getTerminator(); in VerifyShardings()187 Block& function_block = func.front(); in IdentifyXlaShardingForComputationInputs() local189 sharding_for_args.reserve(function_block.getNumArguments()); in IdentifyXlaShardingForComputationInputs()209 llvm::zip(cluster_func.operands(), function_block.getArguments())) { in IdentifyXlaShardingForComputationInputs()357 Block& function_block = func.front(); in IdentifyXlaShardingForComputationOutputs() local358 Operation* terminator = function_block.getTerminator(); in IdentifyXlaShardingForComputationOutputs()471 Block& function_block = func.front(); in IdentifyXlaShardingForTPUComputation() local473 llvm::zip(sharding_for_args, function_block.getArguments())) { in IdentifyXlaShardingForTPUComputation()[all …]
106 auto& function_block = function->getCallableRegion()->front(); in CreateTPUCluster() local107 builder->setInsertionPointToStart(&function_block); in CreateTPUCluster()113 auto& function_body = function_block.getOperations(); in CreateTPUCluster()119 mlir::Operation* function_block_terminator = function_block.getTerminator(); in CreateTPUCluster()