- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 27 for only (0.16 sec)
-
tensorflow/c/eager/custom_device_testutil.cc
tensorflow::string underlying_device; // Set to true whenever a TensorHandle is copied onto the device bool* arrived_flag; // Set to true whenever an operation is executed bool* executed_flag; // If true, only explicit op placements are accepted. If false, uses // type-based dispatch. bool strict_scope_placement; }; struct LoggedTensor { TFE_TensorHandle* tensor; LoggedTensor() = delete;
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/parallel_device/parallel_device_lib.cc
TFE_OpSetCancellationManager(op_.get(), wrap(cancellation_manager_), status); if (TF_GetCode(status) != TF_OK) return; // unwrap op_ and set step_id only if valid step id value was set. // Currently only required for non-TFRT use cases, e.g., EagerOp. if (step_id_.has_value()) { tensorflow::unwrap(op_.get())->SetStepId(step_id_.value()); }
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/eager/parallel_device/parallel_device_testlib.cc
TFE_OpSetAttrString(op.get(), "container", "", 0); // Use the special GUID for no buffer sharing // // TODO(allenl): Should we provide a better API for this? AFAIK this is the // only reasonable way to make variables with no aliasing using the eager C // API. std::string no_sharing = "cd2c89b7-88b7-44c8-ad83-06c2a9158347"; TFE_OpSetAttrString(op.get(), "shared_name", no_sharing.c_str(),
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) -
tensorflow/c/eager/parallel_device/parallel_device.cc
// // 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 // need to return something here or address these use-cases one by one.
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
// 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 // 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
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/experimental/filesystem/modular_filesystem_registration.cc
TF_WRITABLE_FILE_OPS_ABI, "writable file")); if (ops->read_only_memory_region_ops != nullptr) TF_RETURN_IF_ERROR(CheckABI(ops->read_only_memory_region_ops_abi, TF_READ_ONLY_MEMORY_REGION_OPS_ABI, "read only memory region")); return OkStatus(); } // Checks if the plugin and core API numbers match, logging mismatches.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 07 22:08:43 GMT 2023 - 12.8K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem.cc
ops->read_only_memory_region_ops = static_cast<TF_ReadOnlyMemoryRegionOps*>( plugin_memory_allocate(TF_READ_ONLY_MEMORY_REGION_OPS_SIZE)); ops->read_only_memory_region_ops->cleanup = tf_read_only_memory_region::Cleanup; ops->read_only_memory_region_ops->data = tf_read_only_memory_region::Data; ops->read_only_memory_region_ops->length = tf_read_only_memory_region::Length;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Aug 23 06:55:53 GMT 2023 - 46.9K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/math_grad.cc
} ~MatMulGradientFunction() override { for (auto input : forward_inputs_) { if (input) { input->Unref(); } } } private: // TODO(b/174778737): Only hold needed inputs. vector<AbstractTensorHandle*> forward_inputs_; AttrBuilder forward_attrs_; }; class NegGradientFunction : public GradientFunction { public: Status Compute(AbstractContext* ctx,
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 15.2K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_test.cc
/*use_function=*/UseFunction()); ASSERT_EQ(errors::OK, s.code()) << s.message(); } TEST_P(UnifiedAPI, TestUnknownShapeTracing) { if (!UseFunction()) { GTEST_SKIP() << "Tracing only test."; } if (UseMlir()) { // TODO(b/173074167): Remove this. GTEST_SKIP() << "MlirTensor::Shape is not implemented yet."; } AbstractContextPtr ctx(BuildFunction("test_fn"));
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
core::RefCountPtr<AbstractFunction> scoped_func; // Returning null tensors from a tf.function is not supported, so we keep // track of indices in the model's outputs are nullptr in this set. // The FunctionDef only outputs the non-null tensors. We later pad the // function op outputs to have nullptrs at the `null_indices`. absl::flat_hash_set<int> null_indices; { AbstractContextPtr func_ctx(BuildFunction(fn_name));
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)