Home
last modified time | relevance | path

Searched defs:shape_info (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference_test.cc49 GraphShapeInfo shape_info; in TEST() local
113 GraphShapeInfo shape_info; in TEST() local
Dshape_inference.cc99 GraphShapeInfo* shape_info) { in StoreOutputShapes()
137 GraphShapeInfo* shape_info) { in InferShapes()
Dtest_util.cc24 const Graph& graph, const GraphShapeInfo& shape_info, in ShapeAnnotationsMatch()
Dencapsulate_util.cc316 GraphShapeInfo shape_info; in PerformStaticShapeInferenceBeforeEncapsulation() local
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py631 def _wrap_shape(shape_info): argument
641 def _wrap_program_shape(shape_info): argument