HomeSort by relevance Sort by last modified time
    Searched refs:hlo_to_stream_number_ (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
stream_assignment.h 37 absl::flat_hash_map<const HloInstruction*, int> hlo_to_stream_number_; member in class:xla::gpu::StreamAssignment
stream_assignment.cc 29 return hlo_to_stream_number_.contains(&hlo);
33 return FindOrDie(hlo_to_stream_number_, &hlo);
42 InsertOrDie(&hlo_to_stream_number_, hlo, stream_num);

Completed in 662 milliseconds