- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for Kata (0.12 sec)
-
tensorflow/c/eager/unified_api_testutil.h
// Return a tensor handle with given type, values and dimensions. template <class T, TF_DataType datatype> Status TestTensorHandleWithDims(AbstractContext* ctx, const T* data, const int64_t* dims, int num_dims, AbstractTensorHandle** tensor) { std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status(
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 4K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/nn_grad.cc
*/ AbstractTensorHandle* upstream_grad = grad_outputs[0]; DCHECK(upstream_grad); // Recover data format from forward pass for gradient. std::string data_format; TF_RETURN_IF_ERROR(forward_attrs_.Get("data_format", &data_format)); // Grad for A grad_inputs[0] = upstream_grad; grad_inputs[0]->Ref(); // Grad for bias
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_test.cc
ctx.reset(ctx_raw); } AbstractTensorHandlePtr x; { AbstractTensorHandle* x_raw = nullptr; float data[] = {0., 0., 0., 0., 0., 0., 0., 0}; int64_t dim_sizes[] = {2, 4}; Status s = TestTensorHandleWithDims<float, TF_FLOAT>(ctx.get(), data, dim_sizes, 2, &x_raw); ASSERT_EQ(errors::OK, s.code()) << s.message(); x.reset(x_raw);
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) -
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)) { outputs[i] = fn_outputs[i - skipped_indices];
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)