Home
last modified time | relevance | path

Searched defs:src_context (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_driver.cc1118 CUcontext src_context = in SynchronousMemcpyD2D() local
1203 CUcontext src_context = in AsynchronousMemcpyD2D() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dshape_refiner.cc455 InferenceContext* src_context = GetContext(input_edge->src()); in ConstantPartialShape() local