Searched defs:on_device (Results 1 – 6 of 6) sorted by relevance
37 bool on_device() const { in on_device() function
37 int64_t duration_ns, bool on_device, in AddTensorFlowOpEvent()53 int64_t duration_ns, bool on_device, in AddTensorFlowOpEventWithKernelDetails()
36 bool on_device, const OpMetrics& metrics, in ConvertOpMetricsToTfStatsRecord()
72 bool on_device, absl::string_view kernel_name, in AddTensorFlowOpEvent()
83 def on_device(self, device): member in PackedDistributedVariable
2853 auto on_device = b_.AddInstruction( in HandleSingleDevice() local