Searched defs:GpuEventStats (Results 1 – 3 of 3) sorted by relevance
30 struct GpuEventStats { struct33 bool IsKernel() const { return !kernel_details.empty(); } in IsKernel()34 bool IsMemCpy() const { return !memcpy_details.empty(); } in IsMemCpy()36 bool IsXlaOp() const { return !hlo_op_names.empty(); } in IsXlaOp()37 bool IsTfOp() const { return !tf_op_fullname.empty(); } in IsTfOp()40 absl::string_view tf_op_fullname;41 absl::string_view equation;42 absl::string_view tensor_shapes;45 std::vector<absl::string_view> hlo_op_names;46 absl::string_view hlo_module_name;[all …]
30 GpuEventStats::GpuEventStats(const XEventVisitor* event) { in GpuEventStats() function in tensorflow::profiler::GpuEventStats
37 const std::function<void(const GpuEventStats&, KernelReport*)>& in ConvertDeviceTraceXPlaneToKernelReports()