- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 41 for op_registry (0.33 sec)
-
tensorflow/compiler/jit/partially_decluster_pass.cc
continue; } DeviceType device_type(""); TF_RETURN_IF_ERROR( DeviceNameToDeviceType(n->assigned_device_name(), &device_type)); TF_RETURN_IF_ERROR(MemoryTypesForNode(graph.op_registry(), device_type, n->def(), &input_mtypes, &output_mtypes)); for (const Edge* e : n->out_edges()) { Node* dst = e->dst();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 15.7K bytes - Viewed (0) -
tensorflow/cc/framework/scope.cc
Scope Scope::NewRootScope() { Graph* graph = new Graph(OpRegistry::Global()); ShapeRefiner* refiner = new ShapeRefiner(graph->versions(), graph->op_registry()); return Scope(new Impl(graph, new Status, new Impl::NameMap, refiner, /* disable_shape_inference */ false)); } Scope Scope::DisabledShapeInferenceScope() { Graph* graph = new Graph(OpRegistry::Global()); ShapeRefiner* refiner =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:57:22 UTC 2024 - 20.9K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass_test.cc
Status BuildXlaOps(const Scope& s, const FunctionDefLibrary& fdef_lib, std::unique_ptr<Graph>* result) { auto graph = std::make_unique<Graph>(OpRegistry::Global()); TF_RETURN_IF_ERROR(s.ToGraph(graph.get())); FunctionLibraryDefinition flib_def(graph->op_registry(), fdef_lib); // Assign all nodes to the CPU device. static const char* kCpuDevice = "/job:localhost/replica:0/task:0/cpu:0"; for (Node* n : graph->nodes()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference.cc
const tensorflow::FunctionLibraryDefinition* fnlib_def, GraphShapeInfo* shape_info) { ShapeRefiner shape_refiner(graph->versions(), graph->op_registry()); shape_refiner.set_require_shape_inference_fns(false); // TODO(dlibenzi): Verify if it is worth trying to infer shaped within // functions. Some functions can be called at multiple locations with
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc
e->src()->name(), ":", e->src_output(), " -> ", e->dst()->name(), ":", e->dst_input()); } } auto output = std::make_unique<Graph>((*graph)->op_registry()); TF_RETURN_WITH_CONTEXT_IF_ERROR( EncapsulateSubgraphsInFunctions( kXlaClusterIdAttr, **graph, RewriteSubgraph, /*reuse_existing_functions=*/true, &output, flib_def),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.1K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass.cc
DeviceType device_type(""); TF_RETURN_IF_ERROR( DeviceNameToDeviceType(n->assigned_device_name(), &device_type)); TF_RETURN_IF_ERROR(MemoryTypesForNode(root.graph()->op_registry(), device_type, n->def(), &input_mtypes, &output_mtypes)); return output_mtypes; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 24.3K bytes - Viewed (0) -
platforms/documentation/docs/src/snippets/native-binaries/cunit/groovy/libs/cunit/2.1-2/include/CUnit/TestDB.h
* its destruction will be handled by the framework. ppRegistry may not be * NULL (checked by assertion), but *ppRegistry can be NULL (in which case the * function has no effect). Note that *ppRegistry will be set to NULL on return. * * @param ppRegistry Address of a pointer to the registry to destroy (non-NULL). */ CU_EXPORT
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Mon Nov 27 17:53:42 UTC 2023 - 40.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/tftext_utils.cc
} } return failure(); } bool IsTFTextRegistered(const tensorflow::OpRegistry* op_registery) { const std::vector<std::string> kTFTextOps = { "WhitespaceTokenizeWithOffsets", }; for (const auto& iter : kTFTextOps) { if (op_registery->LookUp(iter)) { return true; } } return false; } } // namespace TFL
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.4K bytes - Viewed (0) -
pkg/registry/core/service/ipallocator/controller/repair_test.go
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Sat Sep 30 15:46:06 UTC 2023 - 21.1K bytes - Viewed (0) -
tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc
auto ret0 = ops::_Retval(s.WithOpName("ret0"), arg1, 0); std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); TF_CHECK_OK(s.ToGraph(g.get())); FunctionDef *xla_fdef = fdl.add_function(); TF_CHECK_OK(GraphToFunctionDef(*g, "f3", xla_fdef)); } FunctionLibraryDefinition fld(OpRegistry::Global(), fdl); // Build the XLA computation graph. // "arg0" (T=DT_RESOURCE), "arg1" (T=DT_INT32)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 10.5K bytes - Viewed (0)