Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for UpdatePartitionedCallOpWithNewCallee (0.22 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc

      return success();
    }
    
    // Updates a PartitionedCallOp/StatefulPartitionedCallOp according to the
    // resource-lifted new callee function in lifting_info.
    template <typename CallOpType>
    void UpdatePartitionedCallOpWithNewCallee(
        CallOpType call_op, PartitionedCallLiftingInfo& lifting_info) {
      if (!lifting_info.lifted_callee) return;
      // Replace output resource uses with the aliasing input, so that we can remove
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 55.1K bytes
    - Viewed (0)
Back to top