Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/grappler/
H A Dgrappler.cc263 void TF_DeleteGraphProperties(TF_GraphProperties* graph_properties) { in TF_DeleteGraphProperties()
269 void TF_InferStatically(TF_GraphProperties* graph_properties, in TF_InferStatically()
286 void TF_GetInputPropertiesListSize(TF_GraphProperties* graph_properties, in TF_GetInputPropertiesListSize()
296 void TF_GetOutputPropertiesListSize(TF_GraphProperties* graph_properties, in TF_GetOutputPropertiesListSize()
306 void TF_GetInputPropertiesList(TF_GraphProperties* graph_properties, in TF_GetInputPropertiesList()
325 void TF_GetOutputPropertiesList(TF_GraphProperties* graph_properties, in TF_GetOutputPropertiesList()
H A Dgrappler_test.cc195 TF_GraphProperties* graph_properties = in TEST() local
243 TF_GraphProperties* graph_properties = in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
H A Dsegment.cc516 const grappler::GraphProperties* graph_properties, const Node* node) { in OperationCanBeTranslatedToImplicitBatch()
565 const grappler::GraphProperties* graph_properties, const Node* node) { in OperationHasDynamicNonBatchDimension()
640 const grappler::GraphProperties* graph_properties, const Node* node, in GetClusterBatchSizeForNode()
672 void AddSegmentForNode(const grappler::GraphProperties* graph_properties, in AddSegmentForNode()
841 const grappler::GraphProperties* graph_properties, in SegmentGraph()
H A Dsegment_test.cc63 const grappler::GraphProperties* graph_properties, in RunTest()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A 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()
H A Dscoped_allocator_optimizer.h96 void SetGraphProperties(const GraphProperties& graph_properties) { in SetGraphProperties()
H A Dgraph_optimizer_stage.h63 GraphProperties* graph_properties; member
H A Dgeneric_layout_optimizer_transposer.h76 std::unique_ptr<GraphProperties> graph_properties; member
H A Dremapper.cc99 GraphProperties graph_properties; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_graph_test.cc148 grappler::GraphProperties graph_properties(item); in RunConvertGraph() local
H A Dconvert_graph.cc137 const grappler::GraphProperties& graph_properties, in GetEngineInfo()
H A Dconvert_nodes.cc181 void GetOutputProperties(const grappler::GraphProperties& graph_properties, in GetOutputProperties()
195 void GetInputProperties(const grappler::GraphProperties& graph_properties, in GetInputProperties()
869 const grappler::GraphProperties& graph_properties, in TrtNodeValidator()
6020 const Graph* graph, const grappler::GraphProperties& graph_properties, in ConvertSegmentToGraphDef()
H A Dconvert_nodes_test.cc316 grappler::GraphProperties graph_properties(item); in ConvertToTensorOrWeights() local
393 grappler::GraphProperties graph_properties(item); in TEST_F() local
477 grappler::GraphProperties graph_properties(item); in TEST() local
1462 grappler::GraphProperties graph_properties(item); in RunValidation() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dgraph_properties_test.cc371 GraphProperties graph_properties(item); in RunTestAndValidate() local