Searched defs:on_device (Results 1 – 5 of 5) sorted by relevance
37 bool on_device, in AddTensorFlowOpEvent()53 int64 duration_ns, bool on_device, in AddTensorFlowOpEventWithKernelDetails()
37 bool on_device, const OpMetrics& metrics, in ConvertOpMetricsToTfStatsRecord()
36 bool on_device, absl::string_view kernel_name, in AddTensorFlowOpEvent()
87 def on_device(self, device): member in PackedDistributedVariable
2192 auto on_device = b_.AddInstruction( in HandleSingleDevice() local