Home
last modified time | relevance | path

Searched defs:device_sub_shape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_transfer_manager.cc107 Shape& device_sub_shape = in ReadDynamicShapes() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_xfeed.cc288 Shape& device_sub_shape = in ReadDynamicShapesOnCpu() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtransfer_manager.cc213 Shape& device_sub_shape = in ReadDynamicShapes() local