Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for op_ (0.01 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

                               TF_Status* status) const {
      if (op_ == nullptr) {
        TFE_ContextSetExecutorForThread(context, executor_.get());
        op_.reset(TFE_NewOp(context, operation_name, status));
        if (TF_GetCode(status) != TF_OK) return;
        TFE_OpSetDevice(op_.get(), device_.c_str(), status);
        if (TF_GetCode(status) != TF_OK) return;
      } else {
        TFE_OpReset(op_.get(), operation_name, device_.c_str(), status);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 25.9K bytes
    - Viewed (0)
  2. tensorflow/c/eager/gradients.cc

      return op_->SetAttrInt(attr_name, value);
    }
    absl::Status SetAttrFloat(AbstractOperation* op_, const char* attr_name,
                              float value, ForwardOperation* forward_op_) {
      forward_op_->attrs.Set(attr_name, value);
      return op_->SetAttrFloat(attr_name, value);
    }
    absl::Status SetAttrBool(AbstractOperation* op_, const char* attr_name,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_unified_experimental_graph.cc

        tensorflow::StringPiece s(data, length);
        op_->node_builder.Attr(attr_name, s);
        return absl::OkStatus();
      }
      absl::Status SetAttrInt(const char* attr_name, int64_t value) override {
        op_->node_builder.Attr(attr_name, static_cast<int64_t>(value));
        return absl::OkStatus();
      }
      absl::Status SetAttrFloat(const char* attr_name, float value) override {
        op_->node_builder.Attr(attr_name, value);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
Back to top