Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for hocontrol (0.19 sec)

  1. tensorflow/c/c_api.cc

                                     TF_Operation** control_inputs,
                                     int max_control_inputs) {
      int count = 0;
      for (const auto* edge : oper->node.in_edges()) {
        if (edge->IsControlEdge() && !edge->src()->IsSource()) {
          if (count < max_control_inputs) {
            control_inputs[count] = ToOperation(edge->src());
          }
          ++count;
        }
      }
    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/c_api_function.cc

          output_names_vec.push_back(string(output_names[i]));
        }
      }
    
      // Process control output names.
      std::vector<string> control_output_names_vec;
      if (control_output_names) {
        control_output_names_vec.reserve(ncontrol_outputs);
        for (int i = 0; i < ncontrol_outputs; ++i) {
          control_output_names_vec.push_back(string(control_output_names[i]));
        }
      }
    
      // Compute body nodes.
    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)
  3. tensorflow/c/experimental/gradients/math_grad.cc

      }
    
      Status Compute(AbstractContext* ctx,
                     absl::Span<AbstractTensorHandle* const> grad_outputs,
                     absl::Span<AbstractTensorHandle*> grad_inputs) override {
        // TODO(vnvo2409): Add control dependency
        /* Given upstream grad U and a Log1p op: Y = log(1 + X), the gradients are:
         *
         *    dX = U / (1 + X)
         *
         */
    
        AbstractTensorHandle* upstream_grad = grad_outputs[0];
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 15.2K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_test.cc

      // Check that newly-imported scalar and feed have control deps (neg3 will
      // inherit them from input)
      TF_Operation* control_inputs[100];
      int num_control_inputs = TF_OperationGetControlInputs(
          scalar3, control_inputs, TF_OperationNumControlInputs(scalar3));
      ASSERT_EQ(2, num_control_inputs);
      EXPECT_EQ(feed, control_inputs[0]);
      EXPECT_EQ(feed2, control_inputs[1]);
    
      num_control_inputs = TF_OperationGetControlInputs(
    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)
Back to top