- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 43 for flat (0.14 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/dlpack_test.cc
} else { EXPECT_EQ(dltensor_out->strides[i], dltensor_out->shape[i + 1] * dltensor_out->strides[i + 1]); } } } const float* data_in = static_cast<const float*>(dltensor_in->data); const float* data_out = static_cast<const float*>(dltensor_out->data); for (size_t j = 0; j < num_elements; ++j) { EXPECT_EQ(data_out[j], data_in[j]); } TFE_CallDLManagedTensorDeleter(dlm_out);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Jun 30 03:04:46 GMT 2023 - 4.4K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc
&out_components, status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); ExpectScalarEq<float>(out_components[0].get(), 9.); ExpectScalarEq<float>(out_components[1].get(), 4.); } worker_server1.release(); worker_server2.release(); } } // namespace parallel_device
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) -
tensorflow/c/c_api.cc
float value) { desc->node_builder.Attr(attr_name, value); } void TF_SetAttrFloatList(TF_OperationDescription* desc, const char* attr_name, const float* values, int num_values) { desc->node_builder.Attr(attr_name, ArraySlice<const float>(values, num_values)); }
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) -
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/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/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/custom_device_testutil.cc
custom_device.delete_device = &DeleteLoggingDevice; custom_device.execute = &LoggingDeviceExecute; LoggingDevice* device = new LoggingDevice; device->arrived_flag = arrived_flag; device->executed_flag = executed_flag; device->device_name = name; device->underlying_device = "/job:localhost/replica:0/task:0/device:CPU:0"; device->strict_scope_placement = strict_scope_placement;
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) -
tensorflow/c/eager/gradient_checker.cc
GetDims(theta_tensor, theta_dims.data()); // Initialize auxilary data structures. vector<float> thetaPlus_data(num_elems); vector<float> thetaMinus_data(num_elems); AbstractTensorHandle* f_outputs[1]; // Numerical Grad Check for (int i = 0; i < num_elems; i++) { // Get relative epsilon value float epsilon = theta_data[i] == 0 ? 1e-4 : std::abs(theta_data[i] * 1e-4); AbstractTensorHandlePtr two_eps;
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) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
delete[] static_cast<float*>(data); } // Creates a TFE_TensorHandle with value `v`. TensorHandlePtr FloatTensorHandle(float v, TF_Status* status) { const int num_bytes = sizeof(float); float* values = new float[1]; values[0] = v; std::unique_ptr<TF_Tensor, decltype(&TF_DeleteTensor)> tensor( TF_NewTensor(TF_FLOAT, nullptr, 0, values, num_bytes, &FloatDeallocator,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0)