Searched refs:GetRegionBlockWithIndex (Results 1 – 7 of 7) sorted by relevance
85 Block& block = parallel_execute.GetRegionBlockWithIndex(i); in SinkResourceWritesIntoParallelExecute()115 parallel_execute.GetRegionBlockWithIndex(results_to_remove.index()); in SinkResourceWritesIntoParallelExecute()
98 Block& execute_block = parallel_execute_op.GetRegionBlockWithIndex(i); in ExpandParallelExecuteToIslands()
432 parallel_execute_op.GetRegionBlockWithIndex(0); in CreateParallelExecuteForOutsideCompilation()471 parallel_execute_op.GetRegionBlockWithIndex(1); in CreateParallelExecuteForOutsideCompilation()
321 Block& block = parallel_execute.GetRegionBlockWithIndex(index); in AppendTypes()
510 auto& region = parallel_execute_op->GetRegionBlockWithIndex(core); in BuildParallelExecuteOp()
184 Block& ParallelExecuteOp::GetRegionBlockWithIndex(unsigned index) { in GetRegionBlockWithIndex() function in mlir::tf_device::ParallelExecuteOp191 GetRegionBlockWithIndex(region_index).getTerminator()->getNumOperands(); in GetRegionOutputs()196 GetRegionBlockWithIndex(region_id).getTerminator()->getNumOperands(); in GetRegionOutputs()205 return BlockWrapsSingleOp(&GetRegionBlockWithIndex(index)); in RegionWrapsSingleOp()
162 Block& GetRegionBlockWithIndex(unsigned index);