Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/convert/
H A Dxplane_to_op_stats.cc149 bool is_gpu = device_planes.empty(); in ConvertXSpaceToOpStats() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dgraph_memory.cc175 const bool is_gpu = (device_name.find("GPU:") || device_name.find("gpu:")); in InferFromTrace() local
/aosp_15_r20/external/skia/tests/
H A DSkSLTest.cpp122 static constexpr bool is_gpu(SkSLTestFlags flags) { in is_gpu() function
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dutils.py1799 def is_gpu(device: str): function
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Ddistributed_test.py6145 def _test_ddp_logging_data(self, is_gpu): argument