Home
last modified time | relevance | path

Searched refs:FindMutableTensorCorePlanes (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/utils/
H A Dtpu_xplane_utils.h29 std::vector<XPlane*> FindMutableTensorCorePlanes(XSpace* xspace);
H A Dtpu_xplane_utils.cc34 std::vector<XPlane*> FindMutableTensorCorePlanes(XSpace* xspace) { in FindMutableTensorCorePlanes() function
H A Dtpu_xplane_utils_test.cc49 std::vector<XPlane*> xplanes = FindMutableTensorCorePlanes(&xspace); in TEST()
H A Dderived_timeline.cc383 for (XPlane* plane : FindMutableTensorCorePlanes(space)) { in GenerateDerivedTimeLines()