Home
last modified time | relevance | path

Searched defs:shape_refiner (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference.cc44 ShapeRefiner* shape_refiner) { in PropagateShapes()
98 Status StoreOutputShapes(const Graph& graph, const ShapeRefiner& shape_refiner, in StoreOutputShapes()
138 ShapeRefiner shape_refiner(graph->versions(), graph->op_registry()); in InferShapes() local
Dxla_fusion_optimizer.cc139 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in Optimize() local
Dencapsulate_subgraphs_pass.cc1920 const ShapeRefiner& shape_refiner, in DoStaticShapeInferenceForOutsideCompilationSend()
2343 BackEdgeHelper* back_edge_helper, ShapeRefiner* shape_refiner, in MakeGraphForOutsideCompilationSends()
2393 ShapeRefiner shape_refiner(graph_out->versions(), graph_out->op_registry()); in GetShapeInfoForOutsideCompilationSends() local
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc98 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in LoadGraphFromProto() local
325 Graph* graph, ShapeRefiner* shape_refiner) { in TransformGraphToAddAggregatedInputNode()
423 const ShapeRefiner& shape_refiner, const Node& node, in RegisterNode()
456 void GraphTransferer::RegisterConstantNode(const ShapeRefiner& shape_refiner, in RegisterConstantNode()
605 const ShapeRefiner& shape_refiner) { in IsNodeFlattenReshape()
645 const ShapeRefiner& shape_refiner, const Node& node) { in RegisterNodeWithPaddingAndStrides()
679 const ShapeRefiner& shape_refiner, const Node& node) { in RegisterNodeWithRank()
712 const ShapeRefiner& shape_refiner, const Node& node) { in RegisterPadNode()
786 const ShapeRefiner& shape_refiner, const Node& node) { in RegisterInputNode()
804 const ShapeRefiner& shape_refiner, const Node& node) { in RegisterFlattenNode()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc443 Graph* graph, ShapeRefiner* shape_refiner) { in PropagateShapeInference()
489 const Graph& graph, const ShapeRefiner& shape_refiner, in BuildTensorShapeMapFromGraph()
588 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in BuildAndAddTensorShapes() local
731 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in ClusterizeNodes() local
836 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in BuildClusterSubgraphDef() local
890 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in BuildClusterByBorder() local
961 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in FuseCluster() local
Dremote_fused_graph_execute_utils_test.cc349 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in TEST() local
468 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in TEST() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1759 SymbolicShapeRefiner* shape_refiner, const NodeDef* qnode, in RelaxEnqueueShapesAndMergeTypes()
1783 Status GraphProperties::UpdateMerge(SymbolicShapeRefiner* shape_refiner, in UpdateMerge()
1836 Status GraphProperties::UpdateEnter(SymbolicShapeRefiner* shape_refiner, in UpdateEnter()
1864 SymbolicShapeRefiner* shape_refiner, in UpdateShapes()
1893 SymbolicShapeRefiner* shape_refiner, TopoQueue* new_shapes, in PropagateShapes()
1945 SymbolicShapeRefiner* shape_refiner, in UpdateQueue()
1994 SymbolicShapeRefiner* shape_refiner, bool* new_shapes) { in UpdateEnqueue()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_lib.cc219 ShapeRefiner shape_refiner(input_graph.versions(), input_graph.op_registry()); in FoldConstants() local