Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization_config.cc42 bool ParseInputNodeQuantSpecs(absl::string_view node_names, in ParseInputNodeQuantSpecs()
83 const std::vector<std::string>& node_names, in GetInputNodeQuantSpecs()
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_inspection_required_ops_utils.cc153 std::unordered_set<string>* node_names) { in Uniquify()
169 std::unordered_set<string>* node_names) { in AddInputIdentity()
212 std::unordered_set<string>* node_names) { in AddOutputIdentities()
284 std::unordered_set<string> node_names(graph->num_nodes()); in IsolateNode() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dmlir_roundtrip_flags.cc56 std::vector<string> node_names; in ParseInputArrayInfo() local
66 const std::vector<string>& node_names, in ParseInputArrayInfo()
/external/tensorflow/tensorflow/core/debug/
Ddebug_grpc_testlib.h54 std::vector<string> node_names; variable
Dgrpc_session_debug_test.cc153 string node_names[3]; in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dgraphdef_to_tfl_flatbuffer.cc56 std::vector<string> node_names; in ConvertGraphDefToTFLiteFlatBuffer() local
Dsaved_model_to_tfl_flatbuffer.cc129 std::vector<string> node_names; in ConvertSavedModelToTFLiteFlatBuffer() local
Dtf_tfl_flatbuffer_helpers.cc196 mlir::TFL::QuantizationSpecs* quant_specs, std::vector<string>* node_names, in PopulateQuantizationSpecs()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
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
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgraph_optimizer_stage.cc118 const std::vector<string> node_names, in MakeOptimizedNodeName()
/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.cc156 const string& fn_name, const NodeNameMapping& node_names, in FillFunctionBody()
387 NodeNameMapping node_names; in GraphToFunctionDef() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
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()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond_test.cc152 absl::flat_hash_set<absl::string_view> node_names; in TEST() local
/external/tensorflow/tensorflow/core/grappler/utils/
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
Dgrappler_test.cc117 const GraphDef& graph, const std::vector<string>& node_names, in EvaluateNodes()
Dgraph_view_test.cc139 std::vector<string> node_names = {"a", "b", "c", "d"}; in TYPED_TEST() local
2005 absl::Span<const string> node_names) { in CompareGraphOrder()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmap_vectorization_test.py228 def _make_dataset(node_names): argument
503 def _make_dataset(node_names): argument
/external/tensorflow/tensorflow/python/debug/cli/
Danalyzer_cli.py825 def _exclude_denylisted_ops(self, node_names): argument
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils_test.cc481 const std::unordered_set<string>& node_names = std::get<0>(cluster); in TEST() local
Dremote_fused_graph_execute_utils.cc729 const std::unordered_set<string>& node_names, const GraphDef& graph_def, in ClusterizeNodes()
833 const std::unordered_set<string>& node_names = std::get<0>(cluster); in BuildClusterSubgraphDef() local
/external/tensorflow/tensorflow/core/graph/
Dgraph_test.cc657 std::vector<string> node_names{"_SOURCE", "_SINK", "A", "B", "C"}; in TEST_F() local
/external/tensorflow/tensorflow/python/training/tracking/
Dutil.py100 def node_names(self): member in _ObjectGraphProtoPrettyPrinter
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler_test.cc2320 std::vector<string> node_names; in ValidateNodeDefs() local