- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for hooks (0.16 sec)
-
tensorflow/c/eager/c_api_distributed_test.cc
} TEST(CAPI, DistributedFunctionNoError) { TestDistributedFunctionCancellation(false); } // TODO(b/170399182): Update test once an alternative to using the function // optimization hook is in place. TEST(CAPI, DISABLED_DistributedFunctionCancelledOnError) { TestDistributedFunctionCancellation(true); } void TestRemoteExecuteDeleteContextWithOutstandingRPC(bool async) {
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/c_api_test.cc
EXPECT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); } void BuildExpectedGraph(bool grad_inputs_provided, TF_Output* expected_grad_outputs) { // The expected graph looks like this if grad_inputs_provided. // If grad_inputs_provided is false, Const_0 will be a OnesLike op. // ^ ^ // dy| dx| // MatMul Gradient Graph // | |
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/gradients.cc
// Builds a tensor filled with ones with the same shape and dtype as `t`. Status BuildOnesLike(const TapeTensor& t, AbstractTensorHandle** result) const override; // Looks up the ID of a Gradient. int64_t TensorId(AbstractTensorHandle* tensor) const override; // Converts a Gradient to a TapeTensor. TapeTensor TapeTensorFromGradient(AbstractTensorHandle* g) const override;
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.cc
tensorflow::unwrap(ctx)->SetThreadLocalDevicePlacementPolicy( static_cast<tensorflow::ContextDevicePlacementPolicy>(policy)); } // Note: this function looks up a thread local policy. So it should be called in // the appropriate client thread. In particular, in async mode, it may not be // safe to call this function from the async EagerExecutor threads.
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) -
tensorflow/c/c_api_function_test.cc
// - input name (as it appears in FunctionDef) // - name of output tensor (in nested "add:z:0" format) // dst is either: // - output name (as it appears in FunctionDef) // - <name_of_node>:<index_of_this_input_into_node> (this looks the same as // output tensor naming, but it the index is actually an input index) struct EdgeSpec : public std::pair<string, string> { typedef std::pair<string, string> Base; // Inherit the set of constructors
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 20 22:08:54 GMT 2023 - 63.6K bytes - Viewed (6)