Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 19 for op (0.13 sec)

  1. 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 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  2. tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc

      ParallelDevice parallel_device(std::move(devices));
      std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> handle_op(
          TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp);
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT);
      TFE_OpSetAttrShape(handle_op.get(), "shape", /*dims=*/nullptr, /*num_dims=*/0,
                         status.get());
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 15.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_cluster_test.cc

      TF_Status* status = TF_NewStatus();
      TFE_Op* op = TFE_NewOp(ctx, "ReadVariableOp", status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_OpSetAttrType(op, "dtype", TF_FLOAT);
      TFE_OpAddInput(op, var, status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      int num_retvals = 1;
      TFE_Execute(op, out_value, &num_retvals, status);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Apr 14 10:03:59 GMT 2023
    - 19.3K bytes
    - Viewed (0)
  4. tensorflow/c/eager/gradients.cc

    namespace internal {
    Status Reset(AbstractOperation* op_, const char* op,
                 const char* raw_device_name, ForwardOperation* forward_op_) {
      forward_op_->op_name = op;
      forward_op_->attrs.Reset(op);
      return op_->Reset(op, raw_device_name);
    }
    Status AddInput(AbstractOperation* op_, AbstractTensorHandle* input,
                    ForwardOperation* forward_op_) {
      TF_RETURN_IF_ERROR(op_->AddInput(input));
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  5. tensorflow/c/c_test_util.cc

                        TF_Graph* graph, TF_Status* s, const char* name,
                        TF_Operation** op, const string& op_device, bool check) {
      TF_OperationDescription* desc = TF_NewOperation(graph, op_name, name);
      if (!op_device.empty()) {
        TF_SetDevice(desc, op_device.c_str());
      }
      TF_AddInput(desc, {l, 0});
      TF_AddInput(desc, {r, 0});
      *op = TF_FinishOperation(desc, s);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:52 GMT 2021
    - 17.8K bytes
    - Viewed (2)
  6. 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 30 12:39:09 GMT 2024
    - Last Modified: Tue Jun 15 15:44:44 GMT 2021
    - 12.5K bytes
    - Viewed (0)
  7. 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 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 29.3K bytes
    - Viewed (1)
  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 30 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 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api_distributed_test.cc

          "      name: 'read0'"
          "      op: 'ReadVariableOp'"
          "      input: 'var'"
          "      device: '/job:localhost/replica:0/task:0/device:CPU:0'"
          "      attr {"
          "        key: 'dtype'"
          "        value {"
          "          type: DT_FLOAT"
          "        }"
          "      }"
          "    }"
          "    node_def {"
          "      name: 'read1'"
          "      op: 'ReadVariableOp'"
          "      input: 'var'"
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 23.5K bytes
    - Viewed (0)
Back to top