Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for opr2_type (0.07 sec)

  1. tensorflow/c/eager/c_api_unified_experimental.cc

                               TF_Status* s) {
      unwrap(o)->outputs.push_back(unwrap(tensor));
    }
    
    void TF_AbstractOpSetOpType(TF_AbstractOp* op, const char* const op_type,
                                TF_Status* s) {
      tsl::Set_TF_Status_from_Status(
          s, unwrap(op)->Reset(op_type,
                               /*raw_device_name=*/nullptr));
    }
    
    void TF_AbstractOpSetOpName(TF_AbstractOp* op, const char* const op_name,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 9K bytes
    - Viewed (0)
  2. tensorflow/c/eager/gradients.cc

          gtl::ArraySlice<AbstractTensorHandle*> gradient_tensors) const override;
    
      // Calls the passed-in backward function.
      // op_type is the op's name provided in RecordOperation.
      absl::Status CallBackwardFunction(
          const string& op_type, GradientFunction* gradient_function,
          const std::vector<int64_t>& unneeded_gradients,
          gtl::ArraySlice<AbstractTensorHandle*> output_gradients,
    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/gradients.h

    // `GradientTape::Watch`.) The op_id is simply a unique index assigned to each
    // op executed under the tape. A separate map (`tensorflow::eager::OpTape`)
    // maintains the map from `op_id` to a `OpTapeEntry` which stores the `op_type`,
    // inputs and outputs and the gradient function These data structures combined
    // allow us to trace the data dependencies between operations and hence compute
    // gradients.
    //
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 6.9K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_unified_experimental_graph.cc

      absl::Status SetAttrType(const char* const attr_name,
                               DataType value) override {
        if (!op_) {
          return absl::Status(
              absl::StatusCode::kFailedPrecondition,
              "op_type and op_name must be specified before specifying attrs.");
        }
        op_->node_builder.Attr(attr_name, value);
        return absl::OkStatus();
      }
      absl::Status SetAttrShape(const char* attr_name, const int64_t* dims,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_experimental.cc

      return input_arg.number_attr().c_str();
    }
    
    int TF_OpIsStateful(const char* op_type, TF_Status* status) {
      const tensorflow::OpRegistrationData* op_reg_data;
      status->status =
          tensorflow::OpRegistry::Global()->LookUp(op_type, &op_reg_data);
      if (!status->status.ok()) {
        return 0;
      }
      return op_reg_data->op_def.is_stateful();
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 29.5K bytes
    - Viewed (0)
  6. tensorflow/c/c_api.cc

                                                   const char* op_type,
                                                   const char* oper_name)
        TF_EXCLUSIVE_LOCKS_REQUIRED(graph->mu) {
      return new TF_OperationDescription(graph, op_type, oper_name);
    }
    
    TF_OperationDescription* TF_NewOperation(TF_Graph* graph, const char* op_type,
                                             const char* oper_name) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
Back to top