Searched defs:shape_refiner (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | shape_inference.cc | 48 ShapeRefiner* shape_refiner) { in PropagateShapes()
206 Status StoreOutputShapes(const Graph& graph, const ShapeRefiner& shape_refiner, in StoreOutputShapes()
246 ShapeRefiner shape_refiner(graph->versions(), graph->op_registry()); in InferShapes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | graph_properties.cc | 2236 SymbolicShapeRefiner* shape_refiner, const NodeDef* qnode, in RelaxEnqueueShapesAndMergeTypes()
2260 Status GraphProperties::UpdateMerge(SymbolicShapeRefiner* shape_refiner, in UpdateMerge()
2313 Status GraphProperties::UpdateEnter(SymbolicShapeRefiner* shape_refiner, in UpdateEnter()
2341 SymbolicShapeRefiner* shape_refiner, in UpdateShapes()
2370 SymbolicShapeRefiner* shape_refiner, TopoQueue* new_shapes, in PropagateShapes()
2422 SymbolicShapeRefiner* shape_refiner, in UpdateQueue()
2471 SymbolicShapeRefiner* shape_refiner, bool* new_shapes) { in UpdateEnqueue()
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | fold_constants_lib.cc | 219 ShapeRefiner shape_refiner(input_graph.versions(), input_graph.op_registry()); in FoldConstants() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| H A D | host_compute_ops.cc | 314 ShapeRefiner shape_refiner{graph->versions().producer(), in InferOutputShapes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_device_util.cc | 662 ShapeRefiner shape_refiner(TF_GRAPH_DEF_VERSION, &flib_def); in PrepareGraphForMlir() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | functionalize_cond.cc | 1514 ShapeRefiner shape_refiner{graph_->versions().producer(), in FunctionalizeInternal() local
|