Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dspmd_expander_common.cc348 auto device_return = mlir::cast<mlir::tf_device::ReturnOp>(owner); in TraceUseToNextTFOp() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
H A Dtf_to_tfrt.cc2056 auto device_return = in CreateCompiledModule() local