- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for running (0.18 sec)
-
tensorflow/c/c_api_experimental.cc
(*config.mutable_device_count())["CPU"] = num_cpu_devices; // TODO(b/113217601): This is needed for EagerContext::runner_ to use a // threadpool, so that we avoid the possibility of running the runner_ in the // threadpool of GPU event mgr, as that can trigger more callbacks to be // scheduled on that same threadpool, causing a deadlock in cases where the
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
mutation_type, " after it was run by a session. This mutation will have no effect, " "and will trigger an error in the future. Either don't modify " "nodes after running them or create a new session."); } } } namespace { // Helper method that creates a shape handle for a shape described by dims. tensorflow::shape_inference::ShapeHandle ShapeHandleFromDims(
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/parallel_device/parallel_device_lib.cc
// the status from `TFE_Execute` and returns outputs if the status is OK. std::vector<TensorHandlePtr> Join(TF_Status* status); // Block until all Ops finished running on the thread. void AsyncWait(TF_Status* status); private: void Run(); void Execute(TFE_Context* context, const char* operation_name, std::vector<TFE_TensorHandle*> inputs,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/experimental/grappler/grappler.cc
const TP_OptimizerConfigs tp_configs, const char* device_type) { ConfigList configs; // disable_model_pruning is turned off by default. if (tp_configs.disable_model_pruning == TF_TriState_On) configs.disable_model_pruning = true; else configs.disable_model_pruning = false; // The other configs are turned on by default. CONFIG_TOGGLE(implementation_selector);
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/parallel_device/parallel_device.cc
return nullptr; } // For TFE_CustomDevice::copy_tensor_from_device in the parallel device // registration. // // Currently this is an error, and un-packing ParallelTensors must be performed // explicitly by running a TPUReplicatedOutput operation on the parallel device. // // TODO(allenl): There are some use-cases that are only supported by copying to // host at the moment (e.g. debug print on a tensor, .numpy(), etc.). We either
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
bool GraphErrorInjectionPass::enabled_ = false; // Test to ensure that a registered graph optimization pass is only executed // once (i.e., on the main function side) in running distributed functions. // This test creates a cluster with two workers, create a variable on the // second worker, and run a distributed function (VariableAddFunction) whose ops
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/c_api.cc
TF_Status status; // Let this custom device choose the device to pin this op on if it // implements the pinning function. if (device_.shall_pin_to_this_device != nullptr) { return device_.shall_pin_to_this_device(tensorflow::wrap(op), &status); } return errors::Unimplemented("No custom device pinning implementation."); } private: TFE_Context* context_; TFE_CustomDevice device_; void* info_;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2)