Searched refs:tf_op_fullname (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | tf_op_utils.h | 50 TfOp ParseTfOpFullname(absl::string_view tf_op_fullname);
58 std::string TfOpEventName(absl::string_view tf_op_fullname);
80 inline bool IsOutsideCompilationOp(absl::string_view tf_op_fullname, in IsOutsideCompilationOp() argument
82 if (absl::EndsWith(tf_op_fullname, ":XlaSendToHost")) return true; in IsOutsideCompilationOp()
90 inline bool IsEmbeddingOp(absl::string_view tf_op_fullname) { in IsEmbeddingOp() argument
91 return absl::StrContains(tf_op_fullname, "Embedding"); in IsEmbeddingOp()
|
| D | tf_op_utils.cc | 67 TfOp ParseTfOpFullname(absl::string_view tf_op_fullname) { in ParseTfOpFullname() argument
69 TfOp tf_op = {Category::kUnknown, tf_op_fullname, kUnknownOp}; in ParseTfOpFullname()
71 absl::StrSplit(tf_op_fullname, absl::MaxSplits(':', 1)); in ParseTfOpFullname()
74 if (absl::StartsWithIgnoreCase(tf_op_fullname, "MEMCPYHToD")) { in ParseTfOpFullname()
77 } else if (absl::StartsWithIgnoreCase(tf_op_fullname, "MEMCPYDToH")) { in ParseTfOpFullname()
119 std::string TfOpEventName(absl::string_view tf_op_fullname) { in TfOpEventName() argument
120 return TfOpEventName(ParseTfOpFullname(tf_op_fullname)); in TfOpEventName()
|
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | xplane_to_kernel_stats_db.cc | 48 absl::string_view tf_op_fullname; in ConvertDeviceTraceXPlaneToKernelReports() local
57 tf_op_fullname = stat.StrOrRefValue(); in ConvertDeviceTraceXPlaneToKernelReports()
76 if (!tf_op_fullname.empty()) { in ConvertDeviceTraceXPlaneToKernelReports()
77 tensorflow::profiler::TfOp tf_op = ParseTfOpFullname(tf_op_fullname); in ConvertDeviceTraceXPlaneToKernelReports()
|
| D | op_stats_to_tf_stats_test.cc | 35 XEventBuilder AddTensorFlowOpEvent(std::string&& tf_op_fullname, in AddTensorFlowOpEvent() argument
40 absl::string_view name = on_device ? kernel_name : tf_op_fullname; in AddTensorFlowOpEvent()
47 *plane->GetOrCreateStatMetadata(std::move(tf_op_fullname))); in AddTensorFlowOpEvent()
51 void AddTensorFlowOpEventWithKernelDetails(std::string&& tf_op_fullname, in AddTensorFlowOpEventWithKernelDetails() argument
59 AddTensorFlowOpEvent(std::move(tf_op_fullname), start_timestamp_ns, in AddTensorFlowOpEventWithKernelDetails()
|
| D | xplane_to_op_metrics_db_test.cc | 34 void AddTensorFlowOpEvent(std::string&& tf_op_fullname, in AddTensorFlowOpEvent() argument
38 absl::string_view name = on_device ? kernel_name : tf_op_fullname; in AddTensorFlowOpEvent()
45 *plane->GetOrCreateStatMetadata(std::move(tf_op_fullname))); in AddTensorFlowOpEvent()
|