Home
last modified time | relevance | path

Searched defs:node_names (Results 1 – 25 of 29) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dplacer_inspection_required_ops_utils.cc149 std::unordered_set<string>* node_names) { in Uniquify()
165 std::unordered_set<string>* node_names) { in AddInputIdentity()
204 std::unordered_set<string>* node_names) { in AddOutputIdentities()
272 std::unordered_set<string> node_names(graph->num_nodes()); in IsolateNode() local
H A Dgraph_constructor.cc1061 const std::vector<absl::string_view>& node_names) { in DFS()
1090 std::vector<absl::string_view> node_names; in PrintCycles() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
H A Dquantization_config.cc56 void ParseCustomOpSpecs(absl::string_view node_names, in ParseCustomOpSpecs()
90 bool ParseInputNodeQuantSpecs(absl::string_view node_names, in ParseInputNodeQuantSpecs()
131 const std::vector<std::string>& node_names, in GetInputNodeQuantSpecs()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Dgrpc_session_test.cc89 string node_names[3]; in TEST() local
125 string node_names[3]; in TEST() local
170 string node_names[3]; in TEST() local
201 string node_names[3]; in TEST() local
233 string node_names[3]; in TEST() local
264 string node_names[3]; in TEST() local
291 string node_names[3]; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dgraph_node_util.cc55 auto node_names = in MergeDebugInfo() local
69 auto node_names = in MergeDebugInfo() local
H A Dgraph_test.cc657 std::vector<string> node_names{"_SOURCE", "_SINK", "A", "B", "C"}; in TEST_F() local
H A Dgraph.cc356 const auto& node_names = experimental_debug_info.original_node_names(); in NodeDebugInfo() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dmlir_roundtrip_flags.cc88 std::vector<string> node_names; in ParseInputArrayInfo() local
144 const std::vector<string>& node_names, in ParseInputArrayInfo()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dgraph_view.cc607 absl::flat_hash_map<absl::string_view, int>* node_names, in GetNodeNamesAndPartitionUpdatedNodes()
693 const absl::flat_hash_map<absl::string_view, int>& node_names, in RemovedOrMissingNodeFanoutsWellFormed()
771 const absl::flat_hash_map<absl::string_view, int>& node_names, in CheckNodeNamesAndFanins()
1624 absl::flat_hash_map<absl::string_view, int>* node_names, in ValidateInternal()
1647 absl::flat_hash_map<absl::string_view, int> node_names; in ApplyMutationInternal() local
H A Dgrappler_test.cc117 const GraphDef& graph, const std::vector<string>& node_names, in EvaluateNodes()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
H A Dsegment_test.cc39 const std::set<string>& node_names) { in MakeCandidateFn()
49 const std::set<string>& node_names) { in MakeInputEdgeCandidateFn()
56 const std::set<string>& node_names) { in MakeOutputEdgeCandidateFn()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dgraph_to_functiondef.cc156 const string& fn_name, const NodeNameMapping& node_names, in FillFunctionBody()
387 NodeNameMapping node_names; in GraphToFunctionDef() local
/aosp_15_r20/external/tensorflow/tensorflow/core/debug/
H A Ddebug_grpc_testlib.h54 std::vector<string> node_names; variable
H A Dgrpc_session_debug_test.cc153 string node_names[3]; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/python/
H A Dgraphdef_to_tfl_flatbuffer.cc59 std::vector<string> node_names; in ConvertGraphDefToTFLiteFlatBuffer() local
H A Dsaved_model_to_tfl_flatbuffer.cc136 std::vector<string> node_names; in ConvertSavedModelToTFLiteFlatBuffer() local
H A Djax_to_tfl_flatbuffer.cc132 std::vector<string> node_names; in ConvertJaxToTFLiteFlatBuffer() local
H A Dtf_tfl_flatbuffer_helpers.cc210 std::vector<string>* node_names, std::vector<string>* node_dtypes, in PopulateQuantizationSpecs()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dgraph_optimizer_stage.cc118 const std::vector<string> node_names, in MakeOptimizedNodeName()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dfunctionalize_cond_test.cc151 absl::flat_hash_set<absl::string_view> node_names; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/cli/
H A Danalyzer_cli.py820 def _exclude_denylisted_ops(self, node_names): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dsplit_utils_test.cc172 std::vector<string> node_names; in GetNodeNames() local
H A Dmake_deterministic_test.cc33 std::vector<string> node_names; in GetNodeNames() local
/aosp_15_r20/external/tensorflow/tensorflow/core/platform/
H A Derrors.h453 std::string FormatOriginalNodeLocationForError(const T& node_names, in FormatOriginalNodeLocationForError()
/aosp_15_r20/external/tensorflow/tensorflow/python/checkpoint/
H A Dcheckpoint.py135 def node_names(self): member in ObjectGraphProtoPrettyPrinter

12