Home
last modified time | relevance | path

Searched defs:graph_properties (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc511 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()
Dsegment_test.cc63 const grappler::GraphProperties* graph_properties, in RunTest()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.h99 void SetGraphProperties(const GraphProperties& graph_properties) { in SetGraphProperties()
Dscoped_allocator_optimizer.cc93 Status CheckTypesAndGetShapes(const GraphProperties& graph_properties, in CheckTypesAndGetShapes()
259 GraphDef* graph, const GraphProperties& graph_properties, in GetInputs()
919 GraphProperties graph_properties(item); in Optimize() local
1079 GraphDef* graph, const GraphProperties& graph_properties) { in ProcessGraphDef()
Dgraph_optimizer_stage.h63 GraphProperties* graph_properties; member
Dgeneric_layout_optimizer_transposer.h70 std::unique_ptr<GraphProperties> graph_properties; member
Dlayout_optimizer.cc405 GraphProcessor(const GraphProperties& graph_properties, in GraphProcessor()
495 const GraphProperties& graph_properties; member
2029 const GraphProperties& graph_properties, in DataLayoutOptimizer()
2236 const GraphProperties& graph_properties, in Tune()
2264 GraphProperties graph_properties(item); in Optimize() local
Dremapper.cc87 GraphProperties graph_properties; member
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph_test.cc159 grappler::GraphProperties graph_properties(item); in RunConvertAfterShape() local
Dconvert_graph.cc131 const grappler::GraphProperties& graph_properties, in GetEngineInfo()
Dconvert_nodes_test.cc539 grappler::GraphProperties graph_properties(item); in ConvertToTensorOrWeights() local
556 grappler::GraphProperties graph_properties(item); in TEST_F() local
626 grappler::GraphProperties graph_properties(item); in TEST_F() local
703 grappler::GraphProperties graph_properties(item); in TEST() local
1619 grappler::GraphProperties graph_properties(item); in RunValidation() local
Dconvert_nodes.cc266 void GetOutputProperties(const grappler::GraphProperties& graph_properties, in GetOutputProperties()
280 void GetInputProperties(const grappler::GraphProperties& graph_properties, in GetInputProperties()
1137 const grappler::GraphProperties& graph_properties, in TrtNodeValidator()
6455 const Graph* graph, const grappler::GraphProperties& graph_properties, in ConvertSegmentToGraphDef()