Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dpropagate_device_id_to_function_args.cc130 auto partitioned_call = llvm::cast<mlir::TF::PartitionedCallOp>(op); in PrependDeviceIdToCallsites() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dfunctional_ops.py1127 def partitioned_call(args, function