- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 56 for int (0.27 sec)
-
tensorflow/c/experimental/gradients/array_grad.cc
public: Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override { 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`?
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) -
tensorflow/c/eager/c_api_remote_test.cc
TFE_Op* matmul = MatMulOp(ctx, h0_task1, h1_task1); TFE_OpSetDevice(matmul, remote_device_name, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_TensorHandle* retvals[1]; int num_retvals = 1; TFE_Execute(matmul, &retvals[0], &num_retvals, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TF_Tensor* t = TFE_TensorHandleResolve(retvals[0], status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Aug 12 00:14:22 GMT 2020 - 5.4K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/math_grad_test.cc
{12, 15, 18, 12, 15, 18, 12, 15, 18}, {6, 6, 6, 15, 15, 15, 24, 24, 24}, {6, 15, 24, 6, 15, 24, 6, 15, 24}}; for (int i{}; i < 4; ++i) { bool transpose_a = transpose_a_vals[i]; bool transpose_b = transpose_b_vals[i]; Model MatMulModel = [transpose_a, transpose_b]( AbstractContext* ctx,
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Thu Apr 13 17:32:14 GMT 2023 - 16.3K bytes - Viewed (0) -
tensorflow/c/experimental/grappler/grappler.cc
size_t* lengths, int num_values, void* storage, size_t storage_size, TF_Status* status) { TF_SetStatus(status, TF_OK, ""); const std::vector<std::string>& nodes = reinterpret_cast<const tensorflow::grappler::GrapplerItem*>(item)->fetch; const int len = std::min(num_values, static_cast<int>(nodes.size())); char* p = static_cast<char*>(storage);
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 06 19:12:29 GMT 2023 - 15K bytes - Viewed (1) -
tensorflow/c/eager/gradients.cc
} Status SetAttrStringList(AbstractOperation* op_, const char* attr_name, const void* const* values, const size_t* lengths, int num_values, ForwardOperation* forward_op_) { std::vector<StringPiece> v(num_values); for (int i = 0; i < num_values; ++i) { v[i] = StringPiece(static_cast<const char*>(values[i]), lengths[i]); } forward_op_->attrs.Set(attr_name, v);
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) -
tensorflow/c/eager/dlpack.cc
// data. bool IsValidStrideCompactRowMajorData(int64_t* shape_arr, int64_t* stride_arr, int ndim) { bool valid = true; int64_t expected_stride = 1; for (int i = ndim - 1; i >= 0; --i) { // Empty tensors are always compact regardless of strides. if (shape_arr[i] == 0) return true;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0) -
tensorflow/c/c_test_util.cc
} TF_Tensor* BoolTensor(bool v) { const int num_bytes = sizeof(bool); bool* values = new bool[1]; values[0] = v; return TF_NewTensor(TF_BOOL, nullptr, 0, values, num_bytes, &BoolDeallocator, nullptr); } TF_Tensor* Int8Tensor(const int64_t* dims, int num_dims, const char* values) { int64_t num_values = 1; for (int i = 0; i < num_dims; ++i) { num_values *= dims[i]; }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
tensorflow/c/experimental/gradients/not_differentiable.cc
namespace tensorflow { namespace gradients { Status NotDifferentiableGradientFunction::Compute( AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) { for (int i = 0; i < grad_inputs.size(); i++) { grad_inputs[i] = nullptr; } return OkStatus(); } Status RegisterNotDifferentiable(GradientRegistry* registry, const string& op) {
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Jun 15 01:15:58 GMT 2022 - 1.3K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_test.cc
MatMulOp(context, hcpu, hdevice), TFE_DeleteOp); TFE_OpSetDevice(matmul.get(), name, status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); TFE_TensorHandle* retval; int num_retvals = 1; TFE_Execute(matmul.get(), &retval, &num_retvals, status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); ASSERT_TRUE(executed); TFE_DeleteTensorHandle(retval);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 27 23:39:24 GMT 2020 - 18.4K bytes - Viewed (0) -
tensorflow/c/c_api_function.cc
const TF_Graph* fn_body, const char* fn_name, int ninputs, const TF_Output* inputs, std::vector<OutputTensor>* input_tensors, std::unordered_map<const Node*, std::vector<int>>* input_nodes) TF_EXCLUSIVE_LOCKS_REQUIRED(fn_body->mu) { input_tensors->reserve(ninputs); for (int i = 0; i < ninputs; ++i) { Node* node = inputs[i].oper ? &inputs[i].oper->node : nullptr; int idx = inputs[i].index;
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)