Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 30 for init (0.15 sec)

  1. tensorflow/c/c_api.cc

      for (const auto* edge : oper->node.out_edges()) {
        if (edge->IsControlEdge() && !edge->dst()->IsSink()) {
          ++count;
        }
      }
      return count;
    }
    
    int TF_OperationGetControlOutputs(TF_Operation* oper,
                                      TF_Operation** control_outputs,
                                      int max_control_outputs) {
      int count = 0;
      for (const auto* edge : oper->node.out_edges()) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_unified_experimental.cc

    void TF_OutputListSetNumOutputs(TF_OutputList* o, int num_outputs,
                                    TF_Status* s) {
      unwrap(o)->expected_num_outputs = num_outputs;
      unwrap(o)->outputs.clear();
      unwrap(o)->outputs.resize(num_outputs);
    }
    int TF_OutputListNumOutputs(TF_OutputList* o) {
      return unwrap(o)->outputs.size();
    }
    TF_AbstractTensor* TF_OutputListGet(TF_OutputList* o, int i) {
      return wrap(unwrap(o)->outputs[i]);
    }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  3. tensorflow/c/eager/dlpack_test.cc

    namespace {
    
    void TestHandleFromDLPack(TF_Status* status, TFE_Context* ctx,
                              std::vector<int64_t> shape,
                              std::vector<int64_t> strides) {
      size_t num_elements = 1;
      for (int i = 0; i < static_cast<int32_t>(shape.size()); ++i) {
        num_elements *= shape[i];
      }
      std::vector<float> data(num_elements);
      for (size_t j = 0; j < num_elements; ++j) {
        data[j] = j;
      }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Jun 30 03:04:46 GMT 2023
    - 4.4K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.cc

      ops->filesystem_ops = static_cast<TF_FilesystemOps*>(
          plugin_memory_allocate(TF_FILESYSTEM_OPS_SIZE));
      ops->filesystem_ops->init = tf_posix_filesystem::Init;
      ops->filesystem_ops->cleanup = tf_posix_filesystem::Cleanup;
      ops->filesystem_ops->new_random_access_file =
          tf_posix_filesystem::NewRandomAccessFile;
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Sun Mar 24 20:08:23 GMT 2024
    - 15.8K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_experimental.cc

    int TF_CheckpointReaderHasTensor(TF_CheckpointReader* reader,
                                     const char* name) {
      return reader->HasTensor(name);
    }
    
    const char* TF_CheckpointReaderGetVariable(TF_CheckpointReader* reader,
                                               int index) {
      return reader->variable_list[index].c_str();
    }
    
    int TF_CheckpointReaderSize(TF_CheckpointReader* reader) {
    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)
  6. tensorflow/c/c_api_test.cc

      }
    }
    
    TEST_F(CApiAttributesTest, EmptyList) {
      auto desc = init("list(int)");
      TF_SetAttrIntList(desc, "v", nullptr, 0);
      auto oper = TF_FinishOperation(desc, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      EXPECT_TF_META("v", 0, TF_ATTR_INT, -1);
    }
    
    TEST_F(CApiAttributesTest, Names) {
      auto desc = init("string");
      TF_SetAttrString(desc, "v", "bunny", 5);
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  7. tensorflow/c/c_api_function.cc

        const TF_Graph* fn_body, const char* fn_name, int ninputs,
        const TF_Output* inputs, std::vector<OutputTensor>* input_tensors,
        std::unordered_map<const Node*, std::vector<int>>* input_nodes)
        TF_EXCLUSIVE_LOCKS_REQUIRED(fn_body->mu) {
      input_tensors->reserve(ninputs);
      for (int i = 0; i < ninputs; ++i) {
        Node* node = inputs[i].oper ? &inputs[i].oper->node : nullptr;
        int idx = inputs[i].index;
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 13.6K bytes
    - Viewed (2)
  8. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

      }
      TFE_OpAddAttrs(op_.get(), attributes);
      for (int input_index = 0; input_index < inputs.size(); ++input_index) {
        TFE_OpAddInput(op_.get(), inputs[input_index], status);
        if (TF_GetCode(status) != TF_OK) return;
      }
      std::vector<TFE_TensorHandle*> unwrapped_results(expected_max_outputs);
      int real_num_outputs = expected_max_outputs;
    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/gradient_checker.cc

      for (int32_t i = start; i < end; i += step) {
        (*data)[i] = i;
      }
    }
    
    // Fills out_dims with the dimensions of the given tensor.
    void GetDims(const TF_Tensor* t, int64_t* out_dims) {
      int num_dims = TF_NumDims(t);
      for (int i = 0; i < num_dims; i++) {
        out_dims[i] = TF_Dim(t, i);
      }
    }
    
    // Runs model as is if output is a scalar,
    // else sums the output tensor before returning.
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 7.3K bytes
    - Viewed (0)
  10. tensorflow/c/experimental/next_pluggable_device/c_api.cc

      // by the underlying tensorflow::VariableInfo.
      TF_VariableInfo(int index, const std::string& name, tensorflow::Var* var) {
        var_info = tensorflow::VariableInfo{index, name, var};
      }
    
      tensorflow::VariableInfo var_info{0, "", nullptr};
    };
    
    TF_VariableInfo* TF_CreateVariableInfoFromContext(TF_OpKernelContext* ctx,
                                                      int index,
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jan 09 00:52:04 GMT 2024
    - 13.9K bytes
    - Viewed (1)
Back to top