Searched defs:retval_core_mapping (Results 1 – 1 of 1) sorted by relevance
102 absl::Span<const tpu::ShardingAndIndex> retval_core_mapping, Graph* graph) { in AssignDevicesToArgsAndRetvals()220 std::vector<tpu::ShardingAndIndex>* retval_core_mapping) { in AssignReturnValueToCore()398 std::vector<tpu::ShardingAndIndex> retval_core_mapping( in CompileTFFunctionToHlo() local