Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 16 for push_back (0.28 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/experimental/filesystem/modular_filesystem.cc

        plugin_status.reserve(files.size());
        for (int i = 0; i < files.size(); i++)
          plugin_status.push_back(TF_NewStatus());
        result = ops_->paths_exist(filesystem_.get(), translated_names.data(),
                                   files.size(), plugin_status.data());
        for (int i = 0; i < files.size(); i++) {
          status->push_back(StatusFromTF_Status(plugin_status[i]));
          TF_DeleteStatus(plugin_status[i]);
        }
      }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Sep 06 19:12:29 GMT 2023
    - 23.1K bytes
    - Viewed (0)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. tensorflow/c/eager/c_api_unified_experimental_test.cc

        TFE_TensorHandle* input_eager = TestScalarTensorHandle(eager_ctx, 2.0f);
        func_args.push_back(TF_CreateAbstractTensorFromEagerTensor(input_eager, s));
        ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
        input_eager = TestScalarTensorHandle(eager_ctx, 3.0f);
        func_args.push_back(TF_CreateAbstractTensorFromEagerTensor(input_eager, s));
        ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
      }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri May 19 21:44:52 GMT 2023
    - 39.1K bytes
    - Viewed (0)
  10. 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)
Back to top