Home
last modified time | relevance | path

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

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