Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 10 for push_back (0.27 sec)

  1. tensorflow/c/eager/tape.h

            // GradientTape which uses ones.
            Gradient* zero = input_tensors[target_index].ZerosLike();
            new_zeros.push_back(zero);
            in_grads.push_back(zero);
          } else {
            in_grads.push_back(nullptr);
          }
        } else {
          in_grads.push_back(current_grad->second);
        }
      }
    
      // Avoid infinite recursion. Whichever forward function we run, it'll end up
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 02 12:40:29 GMT 2024
    - 47.2K bytes
    - Viewed (1)
  2. tensorflow/c/c_api_function.cc

        for (int i = 0; i < noutputs; ++i) {
          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]));
        }
      }
    
    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/eager/parallel_device/parallel_device_lib.cc

        components.reserve(underlying_devices_.size());
        for (int j = 0; j < underlying_devices_.size(); ++j) {
          components.push_back(std::move(per_device_output_tensors[j][i]));
        }
        if (expected_output_shapes[i].IsFullyDefined()) {
          per_device_outputs.push_back(ParallelTensor::FromTensorHandles(
              *this, std::move(components),
              absl::Span<const int64_t>(expected_output_shapes[i].dim_sizes()),
    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)
  4. tensorflow/c/eager/gradients.cc

      forward_op_->inputs.push_back(input);
      return absl::OkStatus();
    }
    Status AddInputList(AbstractOperation* op_,
                        absl::Span<AbstractTensorHandle* const> inputs,
                        ForwardOperation* forward_op_) {
      TF_RETURN_IF_ERROR(op_->AddInputList(inputs));
      for (auto input : inputs) {
        forward_op_->inputs.push_back(input);
      }
      return absl::OkStatus();
    }
    
    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/eager/c_api.cc

            lengths_vector.reserve(s_size);
            for (int i = 0; i < s_size; ++i) {
              const string& v = default_value.list().s(i);
              values_vector.push_back(v.data());
              lengths_vector.push_back(v.size());
            }
            TFE_OpSetAttrStringList(op, attr_name, values_vector.data(),
                                    lengths_vector.data(), s_size);
          }
    
          // Int
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  6. tensorflow/c/c_api_experimental.cc

        TF_Function* func = TF_FunctionImportFunctionDef(
            binary_proto_buf.data(), binary_proto_buf.size(), status);
        if (!status->status.ok()) return {};
        ret.push_back(UniqueFuncPtr(func, TF_DeleteFunction));
      }
      return ret;
    }
    
    TF_Tensor* TF_DequeueNamedTensor(TF_Session* session, int tensor_id,
                                     TF_Status* status) {
      assert(session);
    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)
  7. tensorflow/c/eager/c_api_unified_experimental_graph.cc

        }
        auto* t = dyn_cast<GraphTensor>(outputs[0]);
        if (!t) {
          return tensorflow::errors::InvalidArgument(
              "Unable to cast input to GraphTensor");
        }
        inputs_.push_back(t->output_);
        *output = tensorflow::down_cast<TracingTensorHandle*>(outputs[0]);
        return absl::OkStatus();
      }
    
      Status Finalize(OutputList* outputs, AbstractFunction** f) override {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  8. tensorflow/c/eager/c_api_unified_experimental.cc

      return wrap(unwrap(o)->outputs[i]);
    }
    void TF_OutputListPushBack(TF_OutputList* o, TF_AbstractTensor* tensor,
                               TF_Status* s) {
      unwrap(o)->outputs.push_back(unwrap(tensor));
    }
    
    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,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  9. tensorflow/c/c_api.cc

        TF_ImportGraphDefOptions* opts, TF_Operation* oper) {
      opts->opts.control_dependencies.push_back(oper->node.name());
    }
    
    void TF_ImportGraphDefOptionsAddReturnOutput(TF_ImportGraphDefOptions* opts,
                                                 const char* oper_name, int index) {
      opts->tensor_id_data.push_back(oper_name);
      const string& oper_name_str = opts->tensor_id_data.back();
    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)
  10. tensorflow/c/eager/c_api_experimental.cc

              absl::Span<tensorflow::ImmediateExecutionTensorHandle*>(
                  tensorflow::unwrap(handles), *num_handles),
              &result);
          return tensorflow::wrap(result);
        }
        tensor_handles.push_back(
            tensorflow::TensorHandleFromInterface(unwrapped_handle));
      }
      tensorflow::EagerContext* context =
          tensorflow::ContextFromInterface(tensorflow::unwrap(ctx));
      tensorflow::TensorHandle* handle = nullptr;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 35.9K bytes
    - Viewed (3)
Back to top