| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.h | 71 Status MaterializeShapes(const GraphProperties& properties);
74 const GraphProperties& properties);
76 const GraphProperties& properties);
78 const GraphProperties& properties);
80 const GraphProperties& properties);
81 Status MaterializeConstants(const GraphProperties& properties);
83 bool IsFoldable(const NodeDef& node, const GraphProperties* properties) const;
99 const GraphProperties& properties,
102 const GraphProperties& properties,
105 const GraphProperties& properties,
[all …]
|
| D | scoped_allocator_optimizer.h | 32 class GraphProperties; variable
61 const GraphProperties& graph_properties);
99 void SetGraphProperties(const GraphProperties& graph_properties) { in SetGraphProperties()
105 const GraphProperties* graph_properties_;
|
| D | constant_folding.cc | 375 Status ConstantFolding::MaterializeShapes(const GraphProperties& properties) { in MaterializeShapes()
538 bool ExtractShape(const NodeDef& shape_node, const GraphProperties& properties, in ExtractShape()
579 const NodeDef& node, const GraphProperties& properties) { in MaterializeBroadcastGradientArgs()
701 NodeDef* node, const GraphProperties& properties) { in MaterializeReductionIndices()
805 NodeDef* node, const GraphProperties& properties) { in MaterializeConstantValuedNode()
870 NodeDef* node, const GraphProperties& properties) { in MaterializeOutputValues()
897 const GraphProperties& properties) { in MaterializeConstants()
916 const GraphProperties* properties) const { in IsFoldable()
1114 const GraphProperties& properties) { in GetDataTypeFromNodeOrProps()
1547 const GraphProperties& properties, GraphDef* output, in FoldGraph()
[all …]
|
| D | static_schedule.cc | 33 const GraphProperties& properties, const OpLevelCostEstimator& estimator, in PredictExecutionTime()
96 GraphProperties properties(item); in EstimateEarliestExecutionTimes()
167 GraphProperties properties(item); in EstimateRequiredTimes()
|
| D | pin_to_host_optimizer.cc | 92 GraphProperties* properties, in IsNodeOutputPortHostFriendly()
221 Status IsNodeHostCandidate(const GraphView& graph, GraphProperties* properties, in IsNodeHostCandidate()
307 GraphProperties properties(item); in Optimize()
|
| D | layout_optimizer.h | 55 Status Tune(const GrapplerItem& item, const GraphProperties& graph_properties,
|
| D | graph_optimizer_stage_test.cc | 151 GraphProperties properties(item); in TEST_F()
193 GraphProperties properties(item); in TEST_F()
|
| D | graph_optimizer_stage.h | 51 GraphProperties* graph_properties, NodeMap* node_map, in GraphOptimizerContext()
63 GraphProperties* graph_properties;
|
| D | arithmetic_optimizer.h | 135 std::unique_ptr<GraphProperties> graph_properties_;
|
| D | shape_optimizer.cc | 70 GraphProperties properties(item); in Optimize()
|
| D | layout_optimizer.cc | 404 GraphProcessor(const GraphProperties& graph_properties, in GraphProcessor()
469 const GraphProperties& graph_properties_;
478 const GraphProperties& graph_properties, in OptimizeContext()
494 const GraphProperties& graph_properties;
2020 const GraphProperties& graph_properties, in DataLayoutOptimizer()
2227 const GraphProperties& graph_properties, in Tune()
2255 GraphProperties graph_properties(item); in Optimize()
|
| D | scoped_allocator_optimizer.cc | 92 Status CheckTypesAndGetShapes(const GraphProperties& graph_properties, in CheckTypesAndGetShapes()
253 GraphDef* graph, const GraphProperties& graph_properties, in GetInputs()
892 GraphProperties graph_properties(item); in Optimize()
1052 GraphDef* graph, const GraphProperties& graph_properties) { in ProcessGraphDef()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 138 GraphProperties properties(item); in TEST_F()
182 GraphProperties properties(item); in TEST_F()
209 GraphProperties properties(item); in TEST_F()
277 GraphProperties static_properties(item); in TEST_F()
291 GraphProperties dynamic_properties(item); in TEST_F()
323 GraphProperties properties(item); in TEST_F()
347 GraphProperties properties(item); in TEST_F()
380 GraphProperties properties(item); in TEST_F()
407 GraphProperties properties(item); in TEST_F()
425 GraphProperties properties(item); in TEST_F()
[all …]
|
| D | graph_properties.h | 80 class GraphProperties {
83 explicit GraphProperties(const GrapplerItem& item) : item_(item) {} in GraphProperties() function
|
| D | graph_properties.cc | 718 GraphProperties gp(grappler_function_item); in UpdateFunction()
1876 Status GraphProperties::RelaxEnqueueShapesAndMergeTypes( in RelaxEnqueueShapesAndMergeTypes()
1901 Status GraphProperties::UpdateMerge(SymbolicShapeRefiner* shape_refiner, in UpdateMerge()
1954 Status GraphProperties::UpdateEnter(SymbolicShapeRefiner* shape_refiner, in UpdateEnter()
1981 Status GraphProperties::UpdateShapes( in UpdateShapes()
2010 Status GraphProperties::PropagateShapes( in PropagateShapes()
2062 Status GraphProperties::UpdateQueue(const NodeDef* queue_node, in UpdateQueue()
2109 Status GraphProperties::UpdateEnqueue( in UpdateEnqueue()
2156 Status GraphProperties::InferStatically(bool assume_valid_feeds, in InferStatically()
2389 Status GraphProperties::InferDynamically(Cluster* cluster) { in InferDynamically()
[all …]
|
| D | graph_memory.h | 65 GraphProperties* properties, int64* worst_case,
|
| /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()
|
| D | item_wrapper.cc | 194 tensorflow::grappler::GraphProperties properties(*item); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.h | 29 class GraphProperties; variable
157 friend class ::tensorflow::grappler::GraphProperties;
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.h | 44 const grappler::GraphProperties* graph_properties = nullptr;
|
| D | convert_nodes.h | 128 const Graph* graph, const grappler::GraphProperties& graph_properties,
379 TrtNodeValidator(const grappler::GraphProperties& graph_properties,
414 const grappler::GraphProperties& graph_properties_;
|
| D | convert_graph_test.cc | 157 grappler::GraphProperties graph_properties(item); in RunConvertAfterShape()
|
| /external/tensorflow/tensorflow/tools/def_file_filter/ |
| D | symbols_pybind.txt | 251 tensorflow::grappler::GraphProperties::InferStatically
252 tensorflow::grappler::GraphProperties::GetOutputProperties
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 31 class GraphProperties; variable
72 friend class ::tensorflow::grappler::GraphProperties;
646 friend class ::tensorflow::grappler::GraphProperties;
|