| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model.h | 168 struct NodeDef { struct
187 // @return non-nullptr NodeDef that has valid Node or an error argument
|
| D | model.cc | 38 return FilterNodes([](const NodeDef&) { return true; }); in nodes()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| 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()
|
| 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()
|
| 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()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| 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()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | memory_types.h | 24 class NodeDef; variable
|
| D | graph_def_util.h | 28 class NodeDef; variable
|
| D | node_def.proto | 14 message NodeDef { message
|
| D | node_def_util_test.cc | 844 TEST(FormatNodeForErrorTest, NodeDef) { in TEST() argument
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_node_util.h | 27 class NodeDef; variable
|
| 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()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mirror_pad_mode.h | 44 class NodeDef; variable
|
| D | padding.h | 30 class NodeDef; variable
|
| D | equal_graph_def.h | 26 class NodeDef; variable
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_placer.h | 25 class NodeDef; variable
|
| 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()
|
| D | virtual_scheduler.h | 176 const std::unordered_map<const NodeDef*, NodeState>* node_map) { in Init()
|
| 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()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_utils.h | 32 class NodeDef; variable
|
| D | step_stats_collector.h | 34 class NodeDef; variable
|
| /external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/ |
| D | op_handler_selector.h | 26 class NodeDef; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.h | 41 class NodeDef; variable
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/ |
| D | quantize_model_test_base.py | 69 def _contains_op_with_name(self, nodes: Iterable[node_def_pb2.NodeDef],
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | c_api_tfrt.h | 513 const tensorflow::NodeDef NodeDef() { return fallback_attrs_.BuildNodeDef(); } in NodeDef() function
|