Searched defs:on_device (Results 1 – 7 of 7) sorted by relevance
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()
37 bool on_device() const { in on_device() function
83 def on_device(self, device): member in PackedDistributedVariable
560 let on_device = device_tree.contains_key(*f); in collect_status_per_file() localVariable
2853 auto on_device = b_.AddInstruction( in HandleSingleDevice() local