- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for opr2_type (0.07 sec)
-
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) -
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) -
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) -
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) -
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) -
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)