| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.h | 76 Status MaterializeShapes(const GraphProperties& properties);
79 const GraphProperties& properties);
81 const GraphProperties& properties);
83 const GraphProperties& properties);
85 const GraphProperties& properties);
86 Status MaterializeConstants(const GraphProperties& properties);
88 bool IsFoldable(const NodeDef& node, const GraphProperties* properties);
90 const GraphProperties* properties) const;
92 const GraphProperties* properties) const;
108 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 | 389 Status ConstantFolding::MaterializeShapes(const GraphProperties& properties) { in MaterializeShapes()
562 bool ExtractShape(const NodeDef& shape_node, const GraphProperties& properties, in ExtractShape()
603 const NodeDef& node, const GraphProperties& properties) { in MaterializeBroadcastGradientArgs()
726 NodeDef* node, const GraphProperties& properties) { in MaterializeReductionIndices()
830 NodeDef* node, const GraphProperties& properties) { in MaterializeConstantValuedNode()
898 NodeDef* node, const GraphProperties& properties) { in MaterializeOutputValues()
925 const GraphProperties& properties) { in MaterializeConstants()
944 const GraphProperties* properties) { in IsFoldable()
960 const NodeDef& node, const GraphProperties* properties) const { in IsFoldableUncached()
1034 const GraphProperties* properties) const { in MaybeFoldable()
[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()
222 Status IsNodeHostCandidate(const GraphView& graph, GraphProperties* properties, in IsNodeHostCandidate()
308 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 | 137 std::unique_ptr<GraphProperties> graph_properties_;
|
| D | shape_optimizer.cc | 69 GraphProperties properties(item); in Optimize()
|
| D | layout_optimizer.cc | 405 GraphProcessor(const GraphProperties& graph_properties, in GraphProcessor()
470 const GraphProperties& graph_properties_;
479 const GraphProperties& graph_properties, in OptimizeContext()
495 const GraphProperties& graph_properties;
2029 const GraphProperties& graph_properties, in DataLayoutOptimizer()
2236 const GraphProperties& graph_properties, in Tune()
2264 GraphProperties graph_properties(item); in Optimize()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 159 GraphProperties properties(item); in TEST_F()
200 GraphProperties properties(item); in TEST_F()
227 GraphProperties properties(item); in TEST_F()
299 GraphProperties static_properties(item); in TEST_F()
313 GraphProperties dynamic_properties(item); in TEST_F()
345 GraphProperties properties(item); in TEST_F()
369 GraphProperties properties(item); in TEST_F()
402 GraphProperties properties(item); in TEST_F()
429 GraphProperties properties(item); in TEST_F()
447 GraphProperties properties(item); in TEST_F()
[all …]
|
| D | graph_properties.h | 81 class GraphProperties {
84 explicit GraphProperties(const GrapplerItem& item) : item_(item) {} in GraphProperties() function
|
| D | graph_properties.cc | 909 GraphProperties gp(grappler_function_item); in UpdateFunction()
2191 Status GraphProperties::RelaxEnqueueShapesAndMergeTypes( in RelaxEnqueueShapesAndMergeTypes()
2216 Status GraphProperties::UpdateMerge(SymbolicShapeRefiner* shape_refiner, in UpdateMerge()
2269 Status GraphProperties::UpdateEnter(SymbolicShapeRefiner* shape_refiner, in UpdateEnter()
2296 Status GraphProperties::UpdateShapes( in UpdateShapes()
2325 Status GraphProperties::PropagateShapes( in PropagateShapes()
2377 Status GraphProperties::UpdateQueue(const NodeDef* queue_node, in UpdateQueue()
2424 Status GraphProperties::UpdateEnqueue( in UpdateEnqueue()
2471 Status GraphProperties::InferStatically(bool assume_valid_feeds, in InferStatically()
2716 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()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment_test.cc | 63 const grappler::GraphProperties* graph_properties, in RunTest()
391 grappler::GraphProperties static_graph_properties(item); in TEST_F()
423 grappler::GraphProperties static_graph_properties(item); in TEST_F()
452 grappler::GraphProperties static_graph_properties(item); in TEST_F()
481 grappler::GraphProperties static_graph_properties(item); in TEST_F()
519 grappler::GraphProperties static_graph_properties(item); in TEST_F()
548 grappler::GraphProperties static_graph_properties(item); in TEST_F()
577 grappler::GraphProperties static_graph_properties(item); in TEST_F()
|
| D | segment.h | 82 const grappler::GraphProperties* graph_properties,
|
| D | segment.cc | 511 const grappler::GraphProperties* graph_properties, const Node* node) { in OperationCanBeTranslatedToImplicitBatch()
560 const grappler::GraphProperties* graph_properties, const Node* node) { in OperationHasDynamicNonBatchDimension()
635 const grappler::GraphProperties* graph_properties, const Node* node, in GetClusterBatchSizeForNode()
667 void AddSegmentForNode(const grappler::GraphProperties* graph_properties, in AddSegmentForNode()
683 const grappler::GraphProperties* graph_properties, in SegmentGraph()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.h | 30 class GraphProperties; variable
158 friend class ::tensorflow::grappler::GraphProperties;
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 134 const Graph* graph, const grappler::GraphProperties& graph_properties,
392 TrtNodeValidator(const grappler::GraphProperties& graph_properties,
427 const grappler::GraphProperties& graph_properties_;
|
| D | convert_graph_test.cc | 159 grappler::GraphProperties graph_properties(item); in RunConvertAfterShape()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 31 class GraphProperties; variable
72 friend class ::tensorflow::grappler::GraphProperties;
671 friend class ::tensorflow::grappler::GraphProperties;
|
| /external/tensorflow/tensorflow/tools/def_file_filter/ |
| D | symbols_pybind.txt | 269 tensorflow::grappler::GraphProperties::InferStatically
270 tensorflow::grappler::GraphProperties::GetOutputProperties
|