Search Options

Results per page
Sort
Preferred Languages
Advance

Results 31 - 40 of 41 for size_t (0.27 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      // Create two vectors with different lengths
      std::vector<float> size_two_value{1., 2.};
      std::vector<float> size_three_value{1., 2., 3.};
      TensorHandlePtr size_two(
          VectorFloatTensorHandle(size_two_value, status.get()));
      ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
      TensorHandlePtr size_three(
          VectorFloatTensorHandle(size_three_value, status.get()));
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 29.3K bytes
    - Viewed (1)
  2. tensorflow/c/eager/gradient_checker.cc

                             int input_index, bool use_function,
                             AbstractTensorHandle** numerical_grad) {
      vector<AbstractTensorHandle*> theta_inputs(inputs.size());
      for (int i{}; i < inputs.size(); ++i) {
        theta_inputs[i] = inputs[i];
      }
    
      AbstractTensorHandle* theta =
          theta_inputs[input_index];  // parameter we are grad checking
    
    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)
  3. tensorflow/c/eager/custom_device_testutil.cc

      TFE_Execute(op, op_outputs.data(), num_outputs, s);
      TFE_DeleteOp(op);
      if (TF_GetCode(s) != TF_OK) return;
      std::vector<TFE_TensorHandle*> unwrapped_outputs;
      unwrapped_outputs.reserve(op_outputs.size());
      for (auto* handle : op_outputs) {
        unwrapped_outputs.push_back(handle);
      }
      for (int i = 0; i < *num_outputs; ++i) {
        auto logged_tensor = std::make_unique<LoggedTensor>(unwrapped_outputs[i]);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 03 20:47:31 GMT 2021
    - 8.3K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_function_test.cc

      }
    
      void CompareInt32Tensor(const std::vector<int32_t>& expected, TF_Tensor* t) {
        int32_t* data = static_cast<int32_t*>(TF_TensorData(t));
        size_t size = TF_TensorByteSize(t);
        ASSERT_EQ(expected.size() * sizeof(int32_t), size);
        for (int i = 0; i < expected.size(); ++i) {
          ASSERT_EQ(expected[i], data[i]) << "Different data at index " << i;
        }
      }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
  5. tensorflow/c/eager/c_api_experimental_test.cc

        CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
        size_t size = TFE_TensorHandleDeviceMemorySize(copy, status);
        CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
        int64_t dims[] = {2, 2};
        TFE_TensorHandle* copy_aliased = TFE_NewTensorHandleFromDeviceMemory(
            ctx, name, TF_FLOAT, dims, 2, data, size, &Deleter, copy, status);
        CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 03:14:26 GMT 2023
    - 31.5K bytes
    - Viewed (1)
  6. tensorflow/c/eager/c_api_test_util.cc

      TFE_OpSetAttrInt(op, "group_size", group_size);
      TFE_OpSetAttrInt(op, "group_key", 123);
      TFE_OpSetAttrInt(op, "instance_key", 456);
      TFE_OpSetAttrString(op, "merge_op", "Add", 3);
      TFE_OpSetAttrString(op, "final_op", "Id", 2);
      std::vector<int64_t> subdiv_offsets;
      TFE_OpSetAttrIntList(op, "subdiv_offsets", subdiv_offsets.data(),
                           subdiv_offsets.size());
    
      return op;
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
  7. tensorflow/c/eager/unified_api_test.cc

      }
      int64_t dim_sizes[] = {2, 4};
      for (int i = 0; i < shape.dims(); i++) {
        if (shape.dim_size(i) != dim_sizes[i]) {
          return errors::InvalidArgument("Dim ", i, " expected to be of size ",
                                         dim_sizes[i],
                                         " found: ", shape.dim_size(i));
        }
      }
      return absl::OkStatus();
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 6.7K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_remote_test_util.cc

      TFE_ContextOptionsSetDevicePlacementPolicy(opts, TFE_DEVICE_PLACEMENT_SILENT);
      TFE_Context* ctx = TFE_NewContext(opts, status);
      EXPECT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
      TFE_DeleteContextOptions(opts);
    
      TFE_ContextSetServerDef(ctx, 0, serialized.data(), serialized.size(), status);
      EXPECT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Dec 11 22:56:03 GMT 2020
    - 9.1K bytes
    - Viewed (0)
  9. tensorflow/c/eager/unified_api_testutil.cc

        }
        int retvals = outputs.size() - null_indices.size();
        std::vector<AbstractTensorHandle*> fn_outputs(retvals);
        TF_RETURN_IF_ERROR(fn_op->Execute(
            absl::Span<AbstractTensorHandle*>(fn_outputs.data(), fn_outputs.size()),
            &retvals));
        int skipped_indices = 0;
        for (int i = 0; i < outputs.size(); i++) {
          if (!null_indices.contains(i)) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 5.7K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api_test.cc

    }
    TEST(CAPI, TensorHandleSilentCopy) {
      TensorHandleSilentCopy(false, TFE_DEVICE_PLACEMENT_SILENT,
                             TFE_DEVICE_PLACEMENT_SILENT, false);
    }
    TEST(CAPI, TensorHandleSilentCopyAsync) {
      TensorHandleSilentCopy(true, TFE_DEVICE_PLACEMENT_SILENT,
                             TFE_DEVICE_PLACEMENT_SILENT, false);
    }
    TEST(CAPI, TensorHandleSilentCopyLocalPolicy) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
Back to top