- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 22 for pool (0.14 sec)
-
tensorflow/c/c_api_experimental.cc
tensorflow::GetBuildXlaOpsPassFlags(); bool original = flags->tf_xla_enable_lazy_compilation; flags->tf_xla_enable_lazy_compilation = enable; return original; } unsigned char TF_SetTfXlaCpuGlobalJit(unsigned char enable) { tensorflow::MarkForCompilationPassFlags* flags = tensorflow::GetMarkForCompilationPassFlags(); bool original = flags->tf_xla_cpu_global_jit;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/c_api.cc
unsigned char value) { desc->node_builder.Attr(attr_name, static_cast<bool>(value)); } void TF_SetAttrBoolList(TF_OperationDescription* desc, const char* attr_name, const unsigned char* values, int num_values) { std::unique_ptr<bool[]> b(new bool[num_values]); for (int i = 0; i < num_values; ++i) { b[i] = values[i]; }
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/experimental/gradients/custom_gradient_test.cc
namespace tensorflow { namespace gradients { namespace internal { namespace { using std::vector; class CustomGradientTest : public ::testing::TestWithParam<std::tuple<const char*, bool, bool>> { protected: void SetUp() override { TF_StatusPtr status(TF_NewStatus()); TF_SetTracingImplementation(std::get<0>(GetParam()), status.get()); Status s = StatusFromTF_Status(status.get());
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 4.8K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/grad_test_helper.cc
namespace tensorflow { namespace gradients { namespace internal { void CompareNumericalAndAutodiffGradients( Model model, Model grad_model, AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, bool use_function, double abs_error) { auto num_inputs = inputs.size(); std::vector<AbstractTensorHandle*> outputs(num_inputs); auto s = RunModel(grad_model, ctx, inputs, absl::MakeSpan(outputs),
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/array_grad_test.cc
outputs[0] = temp_outputs[1]; temp_outputs[0]->Unref(); return absl::OkStatus(); } class CppGradients : public ::testing::TestWithParam<std::tuple<const char*, bool, bool>> { protected: void SetUp() override { TF_StatusPtr status(TF_NewStatus()); TF_SetTracingImplementation(std::get<0>(GetParam()), status.get()); status_ = StatusFromTF_Status(status.get());
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem_static.cc
#include "tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.h" #include "tensorflow/core/platform/status.h" namespace tensorflow { // Register the POSIX filesystems statically. // Return value will be unused bool StaticallyRegisterLocalFilesystems() { TF_FilesystemPluginInfo info; TF_InitPlugin(&info); Status status = filesystem_registration::RegisterFilesystemPluginImpl(&info); if (!status.ok()) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Sun Mar 24 20:08:23 GMT 2024 - 1.5K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
TFE_CallDLManagedTensorDeleter(dlmt_vptr); } // Checks whether the stride array matches the layout of compact, row-majored // 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.
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/eager/gradients.cc
ForwardOperation* forward_op_) { std::unique_ptr<bool[]> b(new bool[num_values]); for (int i = 0; i < num_values; ++i) { b[i] = values[i]; } forward_op_->attrs.Set(attr_name, gtl::ArraySlice<const bool>(b.get(), num_values)); return op_->SetAttrBoolList(attr_name, values, num_values); }
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/c_api_distributed_test.cc
} return tensorflow::errors::Internal("Graph pass runs for more than once!"); } private: bool first_call_ = true; }; // After the graph pass is registered, it takes effect globally and can affect // other test cases. Define a static variable to switch it on and off. bool GraphErrorInjectionPass::enabled_ = false; // Test to ensure that a registered graph optimization pass is only executed
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/gradient_checker.cc
Status RunAndMaybeSum(AbstractContext* ctx, Model forward, absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle*> outputs, bool use_function) { AbstractTensorHandle* model_outputs[1]; // Run the model. TF_RETURN_IF_ERROR( RunModel(forward, ctx, inputs, model_outputs, use_function));
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)