Searched defs:shape_refiner (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/compiler/jit/ |
D | shape_inference.cc | 44 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
|
D | xla_fusion_optimizer.cc | 139 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in Optimize() local
|
D | encapsulate_subgraphs_pass.cc | 1920 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/ |
D | graph_transferer.cc | 98 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/ |
D | remote_fused_graph_execute_utils.cc | 443 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
|
D | remote_fused_graph_execute_utils_test.cc | 349 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/ |
D | graph_properties.cc | 1759 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/ |
D | fold_constants_lib.cc | 219 ShapeRefiner shape_refiner(input_graph.versions(), input_graph.op_registry()); in FoldConstants() local
|