Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 38 for Op (0.14 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top