- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 38 for Op (0.14 sec)
-
tensorflow/c/eager/c_api_unified_experimental.cc
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, TF_Status* s) { TracingOperation* tracing_op = dyn_cast<TracingOperation>(unwrap(op));
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/nn_grad.cc
GradientFunction* ReluRegisterer(const ForwardOperation& op) { return new ReluGradientFunction(op.outputs); } GradientFunction* SparseSoftmaxCrossEntropyWithLogitsRegisterer( const ForwardOperation& op) { return new SparseSoftmaxCrossEntropyWithLogitsGradientFunction(op.outputs); } GradientFunction* BiasAddRegisterer(const ForwardOperation& op) { return new BiasAddGradientFunction(op.attrs); } } // namespace gradients
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/c_api.cc
delete lib_handle; } TF_Buffer* TF_GetAllOpList() { std::vector<tensorflow::OpDef> op_defs; tensorflow::OpRegistry::Global()->GetRegisteredOps(&op_defs); tensorflow::OpList op_list; for (const auto& op : op_defs) { *(op_list.add_op()) = op; } TF_Buffer* ret = TF_NewBuffer(); TF_CHECK_OK(MessageToBuffer(op_list, ret)); return ret; }
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
} void Variable::GeneralAssignment(const char* op_name, TFE_Context* context, TFE_TensorHandle* value, TF_Status* status) { std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( TFE_NewOp(context, op_name, status), &TFE_DeleteOp); if (TF_GetCode(status) != TF_OK) return; TFE_OpSetAttrType(op.get(), "dtype", type_); TFE_OpAddInput(op.get(), handle_, status);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_testutil.cc
if (TF_GetCode(s) != TF_OK) return; const char* operation_name = TFE_OpGetName(original_op, s); if (TF_GetCode(s) != TF_OK) return; const TFE_OpAttrs* attributes = TFE_OpGetAttrs(original_op); TFE_Op* op(TFE_NewOp(context, operation_name, s)); if (TF_GetCode(s) != TF_OK) return; TFE_OpAddAttrs(op, attributes); TFE_OpSetDevice(op, dev->underlying_device.c_str(), s);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Mar 03 20:47:31 GMT 2021 - 8.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
const std::string final_op("Id"); TFE_OpSetAttrString(op.get(), "final_op", final_op.c_str(), final_op.length()); TFE_OpSetAttrIntList(op.get(), "subdiv_offsets", nullptr, 0); TFE_OpAddInput(op.get(), input, status); if (TF_GetCode(status) != TF_OK) return nullptr; TFE_TensorHandle* result_handle; int num_retvals = 1;
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 29.3K bytes - Viewed (1) -
tensorflow/c/eager/gradients_test.cc
} AbstractOperationPtr check_numerics_op(ctx->CreateOperation()); ForwardOperation forward_op; Status s = Reset(check_numerics_op.get(), "CheckNumerics", /*raw_device_name=*/nullptr, &forward_op); ASSERT_EQ(errors::OK, s.code()) << s.message(); if (isa<TracingOperation>(check_numerics_op.get())) { s = dyn_cast<TracingOperation>(check_numerics_op.get()) ->SetOpName("check_numerics");
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
return; } else if (execution_state_ == ExecutionState::kReadyToExecute) { // op_outputs_ may have been std::moved op_outputs_ = std::vector<TensorHandlePtr>(); Execute(context_, operation_name_, std::move(op_inputs_), attributes_, expected_max_outputs_, &op_outputs_, status_.get()); execution_state_ = ExecutionState::kHasResult; } }
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device.cc
"first un-packed. Got an un-placed op with an input placed on the " "parallel device."); return; } TFE_Context* context = TFE_OpGetContext(original_op, status); if (TF_GetCode(status) != TF_OK) return; const char* operation_name = TFE_OpGetName(original_op, status); if (TF_GetCode(status) != TF_OK) return; const TFE_OpAttrs* attributes = TFE_OpGetAttrs(original_op); NamedParallelDevice* named_device =
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
NodeDef node_def; tensorflow::ImmediateExecutionOperation* op = tensorflow::unwrap(tfe_op); node_def.set_name(op->Name()); node_def.set_op(op->Name()); for (int i = 0; i < num_inputs; ++i) { node_def.add_input("dummy_input"); } OperationFromInterface(op)->Attrs().FillAttrValueMap(node_def.mutable_attr()); const tensorflow::OpRegistrationData* op_reg_data; status->status =
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0)