| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | model.h | 168 struct NodeDef { struct
187 // @return non-nullptr NodeDef that has valid Node or an error argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | memory_optimizer.cc | 236 const std::unordered_map<const NodeDef*, int>& components) { in GetMaxDownstreamComponents()
304 const std::unordered_map<const NodeDef*, int>& components, in AddRecomputeControlDependencyNodes()
398 const std::unordered_map<const NodeDef*, int>& components, in RecomputeSubgraph()
746 std::pair<NodeDef*, NodeDef*>* swap_pair) { in BuildSwapPair()
808 const std::unordered_map<const NodeDef*, Costs::NanoSeconds>& in FindSwapInTrigger()
925 const std::unordered_map<const NodeDef*, Costs::NanoSeconds>& in FindSwapOutTrigger()
984 std::unordered_map<NodeDef*, SwapInfo>* nodes_to_swap) { in IdentifySwappingCandidates()
|
| H A D | static_schedule.cc | 63 std::unordered_map<const NodeDef*, Costs::NanoSeconds>* completion_times) { in EstimateEarliestExecutionTimes()
135 const std::unordered_map<const NodeDef*, Costs::NanoSeconds>& in EstimateRequiredTimes()
|
| H A D | static_schedule_test.cc | 51 const std::unordered_map<const NodeDef*, Costs::NanoSeconds> in GetOrderedTimes()
69 const std::unordered_map<const NodeDef*, Costs::NanoSeconds> in GetOrderedNodeNames()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/ |
| H A D | scc.cc | 58 std::unordered_map<const NodeDef*, int>* components, in StrongConnect()
127 const GraphDef& graph, std::unordered_map<const NodeDef*, int>* components, in StronglyConnectedComponents()
185 std::unordered_map<const NodeDef*, std::vector<int>>* loops) { in IdentifyLoops()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | virtual_scheduler.cc | 171 const std::unordered_map<const NodeDef*, NodeState>* node_map) { in Init()
225 const std::unordered_map<const NodeDef*, NodeState>* node_map, in FirstReadyCmp()
277 const std::unordered_map<const NodeDef*, NodeState>* node_map) { in Init()
323 const std::pair<const NodeDef*, Costs::Duration>& b) { in GetCurrNode()
|
| H A D | virtual_placer.h | 25 class NodeDef; variable
|
| H A D | virtual_scheduler_test.cc | 2358 const std::unordered_set<std::pair<const NodeDef*, int>, in ValidateMemoryUsageSnapshot()
2364 [](const std::pair<const NodeDef*, int>& node_port) { in ValidateMemoryUsageSnapshot()
2697 [](const std::pair<const NodeDef*, int>& node_port) { in TEST_F()
|
| H A D | graph_properties.cc | 2342 const absl::flat_hash_map<const NodeDef*, const NodeDef*>& resource_handles, in UpdateShapes()
2371 const absl::flat_hash_map<const NodeDef*, const NodeDef*>& resource_handles, in PropagateShapes()
2470 const absl::flat_hash_map<const NodeDef*, const NodeDef*>& resource_handles, in UpdateEnqueue()
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/ |
| D | node_def.pb.cc | 47 PROTOBUF_CONSTEXPR NodeDef::NodeDef( in NodeDef() function in tensorflow::NodeDef
336 NodeDef::NodeDef(::PROTOBUF_NAMESPACE_ID::Arena* arena, in NodeDef() function in tensorflow::NodeDef
342 NodeDef::NodeDef(const NodeDef& from) in NodeDef() function in tensorflow::NodeDef
|
| D | node_def.pb.h | 51 class NodeDef; variable
272 inline NodeDef() : NodeDef(nullptr) {} in NodeDef() function
277 NodeDef(NodeDef&& from) noexcept in NodeDef() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | memory_types.h | 24 class NodeDef; variable
|
| H A D | graph_def_util.h | 28 class NodeDef; variable
|
| H A D | node_def.proto | 14 message NodeDef { message
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | graph_node_util.h | 27 class NodeDef; variable
|
| H A D | graph_partition.cc | 813 std::vector<std::pair<const NodeDef*, int64_t>>* nodes, in TopologicalSortNodesWithTimePriority()
814 std::unordered_map<const NodeDef*, int64_t>* node_to_start_time_out) { in TopologicalSortNodesWithTimePriority()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | mirror_pad_mode.h | 44 class NodeDef; variable
|
| H A D | padding.h | 30 class NodeDef; variable
|
| H A D | equal_graph_def.h | 26 class NodeDef; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | function_utils.h | 32 class NodeDef; variable
|
| H A D | step_stats_collector.h | 34 class NodeDef; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/ |
| H A D | op_handler_selector.h | 26 class NodeDef; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | gpu_utils.h | 41 class NodeDef; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/ |
| H A D | quantize_model_test_base.py | 69 def _contains_op_with_name(self, nodes: Iterable[node_def_pb2.NodeDef],
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/ |
| H A D | c_api_tfrt.h | 513 const tensorflow::NodeDef NodeDef() { return fallback_attrs_.BuildNodeDef(); } in NodeDef() function
|