- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 21 for function (0.15 sec)
-
tensorflow/c/c_api_function.cc
for (const Node* n : fn_body->graph.nodes()) { stack_traces[n->name()] = n->GetStackTrace(); } TF_Function* tf_function = new TF_Function(); tf_function->record = new tensorflow::FunctionRecord( std::move(fdef), std::move(stack_traces), false); return tf_function; } TF_Function* TF_GraphToFunction(const TF_Graph* fn_body, const char* fn_name,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 13.6K bytes - Viewed (2) -
tensorflow/c/c_api_experimental.cc
} char* TF_FunctionDebugString(TF_Function* func, size_t* len) { const auto& debug_str = DebugString(func->record->fdef()); *len = debug_str.size(); char* ret = static_cast<char*>(malloc(*len + 1)); memcpy(ret, debug_str.c_str(), *len + 1); return ret; } // On success, returns a set of TF_Function instances from `text_proto` of // GraphDef type. These functions must be deleted by calling TF_DeleteFunction. //
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/experimental/gradients/math_grad_test.cc
/*tfrt*/ ::testing::Values(false), /*use_function*/ ::testing::Values(true, false))); #else INSTANTIATE_TEST_SUITE_P( UnifiedCAPI, CppGradients, ::testing::Combine(::testing::Values("graphdef", "mlir"), /*tfrt*/ ::testing::Values(false), /*use_function*/ ::testing::Values(true, false))); #endif } // namespace } // namespace internal
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Thu Apr 13 17:32:14 GMT 2023 - 16.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
#include "tensorflow/c/eager/parallel_device/parallel_device_testlib.h" #include "tensorflow/c/eager/tfe_context_internal.h" #include "tensorflow/core/common_runtime/eager/context.h" #include "tensorflow/core/framework/function.h" #include "tensorflow/core/lib/core/status_test_util.h" #include "tensorflow/core/platform/test.h" namespace tensorflow { namespace parallel_device { using ::testing::ElementsAre; using ::testing::HasSubstr;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
} // Calls the passed-in backward function. // op_type is the op's name provided in RecordOperation. Status TapeVSpace::CallBackwardFunction( const string& op_type, GradientFunction* gradient_function, const std::vector<int64_t>& unneeded_gradients, gtl::ArraySlice<AbstractTensorHandle*> output_gradients, absl::Span<AbstractTensorHandle*> result) const { if (gradient_function == nullptr) {
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/c_test_util.cc
grads.emplace_back(grad.function_name(), grad.gradient_func()); } std::sort(grads.begin(), grads.end()); return grads; } std::vector<string> GetFuncNames(const tensorflow::GraphDef& graph_def) { std::vector<string> names; auto functions = graph_def.library().function(); names.reserve(functions.size()); for (const tensorflow::FunctionDef& func : functions) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
tensorflow/c/eager/dlpack.cc
default: return tensorflow::errors::InvalidArgument("Unsupported Type Codes: ", dtype.code); } } // Wraps the deleter function of DLManagedTensor to match the function signature // TFE_NewTensorHandleFromDeviceMemory. void DeallocatorWrapperFunc(void* data, size_t len, void* dlmt_vptr) { TFE_CallDLManagedTensorDeleter(dlmt_vptr); }
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/c_api_distributed_test.cc
// 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 // span the local and remote workers. If the graph optimization pass is executed // on both the main function side and the component function side, an error will
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/grappler/grappler.cc
#include "tensorflow/c/tf_buffer.h" #include "tensorflow/c/tf_buffer_internal.h" #include "tensorflow/c/tf_status.h" #include "tensorflow/c/tf_status_helper.h" #include "tensorflow/core/framework/function.h" #include "tensorflow/core/framework/op.h" #include "tensorflow/core/framework/op_def.pb.h" #include "tensorflow/core/grappler/costs/graph_properties.h" #include "tensorflow/core/grappler/costs/op_performance_data.pb.h"
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_test.cc
TF_Operation* operations[]{placeholder_op, reduce_op}; TF_Output y{reduce_op, 0}; const char* output_name = "y"; std::unique_ptr<TF_Function, decltype(&TF_DeleteFunction)> function( TF_GraphToFunction( /* fn_body */ body.get(), /* fn_name */ function_name, /* append_hash_to_fn_name */ 0, /* num_opers */ 2, /* opers */ operations, /* ninputs */ 1, /* inputs */ &x,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 29.3K bytes - Viewed (1)