- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 60 for flat (0.18 sec)
-
tensorflow/c/eager/c_api_unified_experimental.cc
limitations under the License. ==============================================================================*/ #include "tensorflow/c/eager/c_api_unified_experimental.h" #include <vector> #include "absl/container/flat_hash_map.h" #include "absl/strings/str_cat.h" #include "tensorflow/c/eager/c_api_unified_experimental_internal.h" #include "tensorflow/c/tf_datatype.h" #include "tensorflow/c/tf_status.h"
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_testutil.cc
limitations under the License. ==============================================================================*/ #include "tensorflow/c/eager/unified_api_testutil.h" #include "absl/container/flat_hash_set.h" #include "tensorflow/c/eager/c_api_experimental.h" #include "tensorflow/c/eager/c_api_test_util.h" #include "tensorflow/c/eager/c_api_unified_experimental.h"
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) -
tensorflow/c/eager/gradients.h
limitations under the License. ==============================================================================*/ #ifndef TENSORFLOW_C_EAGER_GRADIENTS_H_ #define TENSORFLOW_C_EAGER_GRADIENTS_H_ #include "absl/container/flat_hash_map.h" #include "tensorflow/c/eager/abstract_context.h" #include "tensorflow/c/eager/abstract_tensor_handle.h" #include "tensorflow/c/eager/tape.h" #include "tensorflow/core/common_runtime/eager/attr_builder.h"
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Sep 26 10:27:05 GMT 2022 - 6.9K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
TEST(CAPI, TestFromProto) { Tensor t_cc(DT_FLOAT, TensorShape({2, 3})); t_cc.flat<float>().setConstant(1.0); tensorflow::TensorProto t_proto; t_cc.AsProtoField(&t_proto); TF_Buffer* t_buffer = TF_NewBuffer(); TF_CHECK_OK(MessageToBuffer(t_proto, t_buffer)); const int num_bytes = 6 * sizeof(float); float* values = reinterpret_cast<float*>(tensorflow::cpu_allocator()->AllocateRaw(
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) -
tensorflow/c/eager/BUILD
":tape", "//tensorflow/core/common_runtime/eager:attr_builder", "//tensorflow/core/lib/llvm_rtti", "//tensorflow/core/platform:errors", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/strings", ], ) cc_library( name = "unified_api_testutil", testonly = 1, srcs = [ "unified_api_testutil.cc", ],
Plain Text - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 33.3K bytes - Viewed (0) -
tensorflow/c/eager/gradients_test.cc
ASSERT_EQ(errors::OK, s.code()) << s.message(); ctx.reset(ctx_raw); } AbstractTensorHandlePtr t; { AbstractTensorHandle* x_raw = nullptr; Status s = TestScalarTensorHandle<float, TF_FLOAT>(ctx.get(), 1.0f, &x_raw); ASSERT_EQ(errors::OK, s.code()) << s.message(); t.reset(x_raw); } AbstractOperationPtr check_numerics_op(ctx->CreateOperation()); ForwardOperation forward_op;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7K bytes - Viewed (0) -
tensorflow/c/eager/c_api.h
// be attached. It is intended for use with TFE_OpGetFlatInput to inspect an // already-finalized operation. // // Note that TFE_OpGetFlatInputCount and TFE_OpGetFlatInput operate on a flat // sequence of inputs, unlike TFE_OpGetInputLength (for getting the length of a // particular named input list, which may only be part of the op's inputs). TF_CAPI_EXPORT extern int TFE_OpGetFlatInputCount(const TFE_Op* op,
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 22.8K bytes - Viewed (1) -
tensorflow/c/experimental/next_pluggable_device/c_api.cc
"], which is not type DT_RESOURCE.")); status->status = cc_status; return nullptr; } const tensorflow::ResourceHandle& handle = arg_tensor.flat<tensorflow::ResourceHandle>()(0); tensorflow::Var* variable; cc_status = tensorflow::LookupResource(cc_ctx, handle, &variable); return new TF_VariableInfo(index, handle.name(), variable); }
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 13.9K bytes - Viewed (1) -
tensorflow/c/experimental/gradients/nn_grad_test.cc
auto ReluGradModel = BuildGradModel(ReluModel, registry_); float X_vals[] = {1.0f, 2.0f, 3.0f, -5.0f, -4.0f, -3.0f, 2.0f, 10.0f, -1.0f}; int64_t X_dims[] = {3, 3}; AbstractTensorHandlePtr X; { AbstractTensorHandle* X_raw; status_ = TestTensorHandleWithDims<float, TF_FLOAT>( immediate_execution_ctx_.get(), X_vals, X_dims, 2, &X_raw);
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) -
tensorflow/c/eager/c_api_unified_experimental_test.cc
TF_AbstractTensorGetEagerTensor(result, status.get()); ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); TF_Tensor* result_tensor = TFE_TensorHandleResolve(result_t, status.get()); float* result_value = static_cast<float*>(TF_TensorData(result_tensor)); EXPECT_EQ(*result_value, 4.0); TF_DeleteTensor(result_tensor); TF_DeleteAbstractTensor(result); TF_DeleteOutputList(o); TF_DeleteExecutionContext(ctx);
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)