| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.h | 64 Status MaterializeShapes(const GraphProperties& properties);
67 const GraphProperties& properties);
69 const GraphProperties& properties);
71 const GraphProperties& properties);
72 Status MaterializeConstants(const GraphProperties& properties);
90 const GraphProperties& properties,
93 const GraphProperties& properties,
97 const GraphProperties& properties,
106 const GraphProperties& properties) const;
108 GraphProperties* properties,
[all …]
|
| D | scoped_allocator_optimizer.h | 28 class GraphProperties; variable
55 const GraphProperties& graph_properties);
85 void SetGraphProperties(const GraphProperties& graph_properties) { in SetGraphProperties()
91 const GraphProperties* graph_properties_;
|
| D | constant_folding.cc | 145 bool GetConcatAxis(const GraphProperties& properties, NodeDef* node, in GetConcatAxis()
345 Status ConstantFolding::MaterializeShapes(const GraphProperties& properties) { in MaterializeShapes()
508 bool ExtractShape(const NodeDef& shape_node, const GraphProperties& properties, in ExtractShape()
549 const NodeDef& node, const GraphProperties& properties) { in MaterializeBroadcastGradientArgs()
671 NodeDef* node, const GraphProperties& properties) { in MaterializeReductionIndices()
775 NodeDef* node, const GraphProperties& properties) { in MaterializeConstantValuedNode()
840 const GraphProperties& properties) { in MaterializeConstants()
1010 const GraphProperties& properties) { in GetDataTypeFromNodeOrProps()
1499 const NodeDef& node, const GraphProperties& properties) const { in IsSimplifiableReshape()
1642 int input_to_forward, const GraphProperties& properties, NodeDef* node, in ReplaceOperationWithIdentity()
[all …]
|
| D | static_schedule.cc | 31 const GraphProperties& properties, const OpLevelCostEstimator& estimator, in PredictExecutionTime()
94 GraphProperties properties(item); in EstimateEarliestExecutionTimes()
162 GraphProperties properties(item); in EstimateRequiredTimes()
|
| D | pin_to_host_optimizer.cc | 90 GraphProperties* properties, in IsNodeOutputPortHostFriendly()
218 Status IsNodeHostCandidate(const GraphView& graph, GraphProperties* properties, in IsNodeHostCandidate()
313 GraphProperties properties(item); in Optimize()
|
| D | graph_optimizer_stage_test.cc | 151 GraphProperties properties(item); in TEST_F()
193 GraphProperties properties(item); in TEST_F()
|
| D | layout_optimizer.h | 53 Status Tune(const GrapplerItem& item, const GraphProperties& graph_properties,
|
| D | graph_optimizer_stage.h | 51 GraphProperties* graph_properties, NodeMap* node_map, in GraphOptimizerContext()
63 GraphProperties* graph_properties;
|
| D | arithmetic_optimizer.h | 133 std::unique_ptr<GraphProperties> graph_properties_;
|
| D | shape_optimizer.cc | 35 GraphProperties properties(item); in Optimize()
|
| D | layout_optimizer.cc | 397 GraphProcessor(const GraphProperties& graph_properties, in GraphProcessor()
462 const GraphProperties& graph_properties_;
471 const GraphProperties& graph_properties, in OptimizeContext()
485 const GraphProperties& graph_properties;
1962 const GraphProperties& graph_properties, in DataLayoutOptimizer()
2182 const GraphProperties& graph_properties, in Tune()
2212 GraphProperties graph_properties(item); in Optimize()
|
| D | scoped_allocator_optimizer.cc | 78 Status CheckTypesAndGetShapes(const GraphProperties& graph_properties, in CheckTypesAndGetShapes()
683 GraphProperties graph_properties(item); in Optimize()
818 GraphDef* graph, const GraphProperties& graph_properties) { in ProcessGraphDef()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 128 GraphProperties properties(item); in TEST_F()
172 GraphProperties properties(item); in TEST_F()
199 GraphProperties properties(item); in TEST_F()
261 GraphProperties static_properties(item); in TEST_F()
275 GraphProperties dynamic_properties(item); in TEST_F()
307 GraphProperties properties(item); in TEST_F()
331 GraphProperties properties(item); in TEST_F()
364 GraphProperties properties(item); in TEST_F()
391 GraphProperties properties(item); in TEST_F()
409 GraphProperties properties(item); in TEST_F()
[all …]
|
| D | graph_properties.h | 78 class GraphProperties {
81 explicit GraphProperties(const GrapplerItem& item) : item_(item) {} in GraphProperties() function
|
| D | graph_properties.cc | 688 GraphProperties gp(grappler_function_item); in UpdateFunction()
1758 Status GraphProperties::RelaxEnqueueShapesAndMergeTypes( in RelaxEnqueueShapesAndMergeTypes()
1783 Status GraphProperties::UpdateMerge(SymbolicShapeRefiner* shape_refiner, in UpdateMerge()
1836 Status GraphProperties::UpdateEnter(SymbolicShapeRefiner* shape_refiner, in UpdateEnter()
1863 Status GraphProperties::UpdateShapes( in UpdateShapes()
1892 Status GraphProperties::PropagateShapes( in PropagateShapes()
1944 Status GraphProperties::UpdateQueue(const NodeDef* queue_node, in UpdateQueue()
1991 Status GraphProperties::UpdateEnqueue( in UpdateEnqueue()
2038 Status GraphProperties::InferStatically(bool assume_valid_feeds, in InferStatically()
2253 Status GraphProperties::InferDynamically(Cluster* cluster) { in InferDynamically()
[all …]
|
| D | graph_memory.h | 65 GraphProperties* properties, int64* worst_case,
|
| D | virtual_scheduler.h | 350 std::unique_ptr<GraphProperties> graph_properties_; // Initialized in Init().
|
| D | graph_memory.cc | 73 const std::vector<const NodeDef*>& nodes, GraphProperties* properties, in InferMemUsageForNodes()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.h | 38 TrtCandidateSelector(const grappler::GraphProperties& graph_properties,
52 const grappler::GraphProperties& graph_properties_;
66 const grappler::GraphProperties* graph_properties = nullptr;
|
| D | convert_nodes.h | 136 const Graph* graph, const grappler::GraphProperties& graph_properties,
386 const grappler::GraphProperties& graph_properties);
403 const grappler::GraphProperties& graph_properties,
|
| D | convert_graph_test.cc | 98 grappler::GraphProperties graph_properties(item); in TEST()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | model_analyzer.h | 27 class GraphProperties; variable
37 void PrintNodeInfo(const NodeDef* node, const GraphProperties& properties,
|
| D | model_analyzer.cc | 31 GraphProperties properties(item_); in GenerateReport()
45 const GraphProperties& properties, bool debug, in PrintNodeInfo()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.h | 29 class GraphProperties; variable
187 friend class ::tensorflow::grappler::GraphProperties;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 33 class GraphProperties; variable
76 friend class ::tensorflow::grappler::GraphProperties;
681 friend class ::tensorflow::grappler::GraphProperties;
|