Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 31 for take (0.17 sec)

  1. tensorflow/c/c_api.cc

                                    const char* attr_name, const void* proto,
                                    size_t proto_len, TF_Status* status) {
      // shape.ParseFromArray takes an int as length, this function takes size_t,
      // make sure there is no information loss.
      if (proto_len > std::numeric_limits<int>::max()) {
        status->status = InvalidArgument(
            "proto_len (", proto_len,
    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/parallel_device/parallel_device_lib_test.cc

          ParallelTensor::FromTensorHandles(parallel_device,
                                            std::move(mixed_handles), status.get());
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK);
      // Can't take the shape of a parallel tensor with varying numbers of axes, but
      // running operations on them is OK.
      TF_ASSERT_OK(unknown_length_vector->Shape(&shape));
      EXPECT_THAT(*shape, ElementsAre(-1));
    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/c_api_experimental.cc

      // Update the vector with information from `input_tensors` if provided.
      if (input_tensors != nullptr) {
        // Note that we take the address of the elements in `all_input_tensors`
        // below. Allocate enough space so that no reallocation happens, which will
        // make the pointers invalid.
        all_input_tensors.reserve(num_inputs);
        for (int i = 0; i < num_inputs; ++i) {
    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)
  4. tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc

      TensorHandlePtr combined_value = CreatePerDeviceValues(
          context.get(), in_components, device_name, status.get());
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
    
      // Loop to make synchronization failures more deterministic
      for (int i = 0; i < 100; ++i) {
        TensorHandlePtr multiply_result(
            Multiply(context.get(), combined_value.get(), combined_value.get(),
                     status.get()));
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 22:09:57 GMT 2023
    - 6.7K bytes
    - Viewed (0)
  5. tensorflow/c/experimental/gradients/nn_grad_test.cc

    #include "tensorflow/c/eager/c_api_test_util.h"
    #include "tensorflow/c/eager/unified_api_testutil.h"
    #include "tensorflow/c/experimental/gradients/grad_test_helper.h"
    #include "tensorflow/c/experimental/gradients/tape/tape_context.h"
    #include "tensorflow/c/experimental/ops/nn_ops.h"
    #include "tensorflow/c/tf_status_helper.h"
    #include "tensorflow/core/platform/tensor_float_32_utils.h"
    #include "tensorflow/core/platform/test.h"
    
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 8.3K bytes
    - Viewed (0)
  6. tensorflow/c/experimental/gradients/array_grad.cc

        for (int i = 0; i < grad_outputs.size(); i++) {
          auto grad_input = grad_outputs[i];
          // TODO(srbs): Should we add a copy contructor to AbstractTensorHandle
          // that takes care of this similar to `Tensor`?
          if (grad_input) {
            grad_input->Ref();
          }
          grad_inputs[i] = grad_input;
        }
        return absl::OkStatus();
      }
      ~IdentityNGradientFunction() override {}
    C++
    - Registered: Tue Apr 09 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 1.6K bytes
    - Viewed (0)
  7. tensorflow/c/eager/gradient_checker.cc

        theta_inputs[input_index] = thetaMinus.get();
        TF_RETURN_IF_ERROR(
            RunAndMaybeSum(ctx, forward, theta_inputs, f_outputs, use_function));
        AbstractTensorHandlePtr fMinus(f_outputs[0]);
    
        // Take Difference of both estimates: (f(theta + eps) - f(theta - eps)).
        TF_RETURN_IF_ERROR(
            ops::Sub(ctx, fPlus.get(), fMinus.get(), f_outputs, "sub_top"));
        AbstractTensorHandlePtr fDiff(f_outputs[0]);
    
    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)
  8. tensorflow/c/experimental/filesystem/modular_filesystem_test.cc

      // example they might require the root directory path to be in a special
      // format, etc. When we get there, we might decide to move this class to
      // `modular_filesystem_test.h` and extend the instantiation to also take as
      // argument an implementation for this method/a subclass factory (see
      // https://github.com/google/googletest/blob/master/googletest/docs/advanced.md#creating-value-parameterized-abstract-tests)
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri May 27 20:25:58 GMT 2022
    - 71K bytes
    - Viewed (0)
  9. tensorflow/c/experimental/gradients/tape/tape_operation.cc

    ==============================================================================*/
    #include "tensorflow/c/experimental/gradients/tape/tape_operation.h"
    
    #include "tensorflow/c/eager/abstract_context.h"
    #include "tensorflow/c/eager/gradients.h"
    
    namespace tensorflow {
    namespace gradients {
    TapeOperation::TapeOperation(AbstractOperation* parent_op, Tape* tape,
                                 const GradientRegistry& registry)
        : AbstractOperation(kTape),
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jun 07 01:53:35 GMT 2022
    - 9K bytes
    - Viewed (1)
  10. tensorflow/c/c_api_test.cc

      EXPECT_EQ(string("fake"), string(src_names[0]));
      EXPECT_EQ(0, src_indexes[0]);
    
      TF_DeleteImportGraphDefResults(results);
      TF_DeleteImportGraphDefOptions(opts);
      TF_DeleteBuffer(graph_def);
      TF_DeleteGraph(graph);
      TF_DeleteStatus(s);
    }
    
    TEST(CAPI, Session) {
      TF_Status* s = TF_NewStatus();
      TF_Graph* graph = TF_NewGraph();
    
      // Make a placeholder operation.
    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