Home
last modified time | relevance | path

Searched defs:on_device (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/convert/
Dop_stats_to_tf_stats_test.cc37 bool on_device, in AddTensorFlowOpEvent()
53 int64 duration_ns, bool on_device, in AddTensorFlowOpEventWithKernelDetails()
Dop_stats_to_tf_stats.cc37 bool on_device, const OpMetrics& metrics, in ConvertOpMetricsToTfStatsRecord()
Dxplane_to_op_metrics_db_test.cc36 bool on_device, absl::string_view kernel_name, in AddTensorFlowOpEvent()
/external/tensorflow/tensorflow/python/distribute/
Dpacked_distributed_variable.py87 def on_device(self, device): member in PackedDistributedVariable
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2192 auto on_device = b_.AddInstruction( in HandleSingleDevice() local