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