Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for EagerOperation (0.13 sec)

  1. tensorflow/compiler/mlir/tfr/integration/node_expansion_pass.cc

                                    "The number of nodes being op expanded.");
    }  // namespace
    
    namespace tfr {
    
    Status CompositeOpExpansion::Run(EagerOperation* orig_op,
                                     std::unique_ptr<EagerOperation>* out_op) {
      if (!IsEnabled()) return absl::OkStatus();
      // This can be the default cpu device.
      if (orig_op->Device() != kVariantDeviceNull) return absl::OkStatus();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Feb 25 16:22:36 UTC 2024
    - 3.8K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.cc

        return;
      }
      if (op == nullptr) {
        status->status = tensorflow::errors::InvalidArgument(
            "Got a null or uninitialized `op` argument");
        return;
      }
      tensorflow::EagerOperation* operation =
          OperationFromInterface(tensorflow::unwrap(const_cast<TFE_Op*>(op)));
      operation->MutableAttrs()->Set(attr_name, attr_value);
    }
    
    TF_CAPI_EXPORT extern int TFE_OpGetInputLength(TFE_Op* op,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 08:11:23 UTC 2024
    - 44K bytes
    - Viewed (0)
Back to top