Home
last modified time | relevance | path

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

/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/compiler/jit/
Dshape_inference.cc48 ShapeRefiner* shape_refiner) { in PropagateShapes()
209 Status StoreOutputShapes(const Graph& graph, const ShapeRefiner& shape_refiner, in StoreOutputShapes()
249 ShapeRefiner shape_refiner(graph->versions(), graph->op_registry()); in InferShapes() local
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc444 Graph* graph, ShapeRefiner* shape_refiner) { in PropagateShapeInference()
490 const Graph& graph, const ShapeRefiner& shape_refiner, in BuildTensorShapeMapFromGraph()
589 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in BuildAndAddTensorShapes() local
732 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in ClusterizeNodes() local
838 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in BuildClusterSubgraphDef() local
892 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in BuildClusterByBorder() local
963 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in FuseCluster() local
Dremote_fused_graph_execute_utils_test.cc348 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in TEST() local
467 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in TEST() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc2192 SymbolicShapeRefiner* shape_refiner, const NodeDef* qnode, in RelaxEnqueueShapesAndMergeTypes()
2216 Status GraphProperties::UpdateMerge(SymbolicShapeRefiner* shape_refiner, in UpdateMerge()
2269 Status GraphProperties::UpdateEnter(SymbolicShapeRefiner* shape_refiner, in UpdateEnter()
2297 SymbolicShapeRefiner* shape_refiner, in UpdateShapes()
2326 SymbolicShapeRefiner* shape_refiner, TopoQueue* new_shapes, in PropagateShapes()
2378 SymbolicShapeRefiner* shape_refiner, in UpdateQueue()
2427 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
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dhost_compute_ops.cc298 ShapeRefiner shape_refiner{graph->versions().producer(), in InferOutputShapes() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond.cc1521 ShapeRefiner shape_refiner{graph_->versions().producer(), in FunctionalizeInternal() local