| /external/tensorflow/tensorflow/core/framework/ |
| D | memory_types.h | 24 class NodeDef; variable
|
| D | node_def.proto | 11 message NodeDef { message
|
| D | node_def_util.h | 35 class NodeDef; variable
|
| D | node_def_util_test.cc | 568 TEST(FormatNodeForErrorTest, NodeDef) { in TEST() argument
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | static_schedule.cc | 61 std::unordered_map<const NodeDef*, Costs::NanoSeconds>* completion_times) { in EstimateEarliestExecutionTimes()
130 const std::unordered_map<const NodeDef*, Costs::NanoSeconds>& in EstimateRequiredTimes()
|
| D | memory_optimizer.cc | 216 const std::unordered_map<const NodeDef*, int>& components) { in GetMaxDownstreamComponents()
284 const std::unordered_map<const NodeDef*, int>& components, in AddRecomputeControlDependencyNodes()
378 const std::unordered_map<const NodeDef*, int>& components, in RecomputeSubgraph()
715 std::pair<NodeDef*, NodeDef*>* swap_pair) { in BuildSwapPair()
777 const std::unordered_map<const NodeDef*, Costs::NanoSeconds>& in FindSwapInTrigger()
894 const std::unordered_map<const NodeDef*, Costs::NanoSeconds>& in FindSwapOutTrigger()
951 std::unordered_map<NodeDef*, SwapInfo>* nodes_to_swap) { in IdentifySwappingCandidates()
|
| /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/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/grappler/costs/ |
| D | virtual_placer.h | 23 class NodeDef; variable
|
| D | virtual_scheduler.h | 136 const std::unordered_map<const NodeDef*, NodeState>* node_state) {} in Init()
147 void Init(const std::unordered_map<const NodeDef*, NodeState>* node_state) in Init()
169 void Init(const std::unordered_map<const NodeDef*, NodeState>* node_state) in Init()
|
| D | virtual_scheduler.cc | 94 const std::unordered_map<const NodeDef*, NodeState>* node_state) { in Init()
150 const std::unordered_map<const NodeDef*, NodeState>* node_state) { in Init()
195 const std::pair<const NodeDef*, Costs::Duration>& b) { in GetCurrNode()
|
| D | virtual_scheduler_test.cc | 1677 const std::unordered_set<std::pair<const NodeDef*, int>, in ValidateMemoryUsageSnapshot()
1683 [](const std::pair<const NodeDef*, int>& node_port) { in ValidateMemoryUsageSnapshot()
2471 [](const std::pair<const NodeDef*, int>& node_port) { in TEST_F()
|
| D | graph_properties.cc | 1865 const std::unordered_map<const NodeDef*, const NodeDef*>& resource_handles, in UpdateShapes()
1894 const std::unordered_map<const NodeDef*, const NodeDef*>& resource_handles, in PropagateShapes()
1993 const std::unordered_map<const NodeDef*, const NodeDef*>& resource_handles, in UpdateEnqueue()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.h | 34 class NodeDef; variable
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 785 const GraphDef* gdef, std::vector<std::pair<const NodeDef*, int64>>* nodes, in TopologicalSortNodesWithTimePriority()
786 std::unordered_map<const NodeDef*, int64>* node_to_start_time_out) { in TopologicalSortNodesWithTimePriority()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 249 Status FunctionLibraryRuntimeOverlay::CreateKernel(const NodeDef&, OpKernel**) { in CreateKernel()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 937 const std::vector<std::pair<const NodeDef*, int>>& input_node_and_ports, in ValidateNode()
|