Home
last modified time | relevance | path

Searched refs:GraphShapeInfo (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference.h38 typedef std::unordered_map<string, std::vector<InferredShape>> GraphShapeInfo; typedef
45 GraphShapeInfo* shape_info);
Dshape_inference_test.cc51 GraphShapeInfo shape_info; in TEST()
115 GraphShapeInfo shape_info; in TEST()
161 GraphShapeInfo shape_info; in TEST()
Dtest_util.h41 const Graph& graph, const GraphShapeInfo& shape_info,
Dtest_util.cc24 const Graph& graph, const GraphShapeInfo& shape_info, in ShapeAnnotationsMatch()
Dshape_inference.cc210 GraphShapeInfo* shape_info) { in StoreOutputShapes()
248 GraphShapeInfo* shape_info) { in InferShapes()
Dencapsulate_util.cc326 GraphShapeInfo shape_info; in PerformStaticShapeInferenceBeforeEncapsulation()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.h294 const GraphShapeInfo& shape_info, const Node& node,
557 const GraphShapeInfo& shape_info,
Ddistributed_tpu_rewrite_pass.cc1679 static Status GetEdgeShape(const GraphShapeInfo& shape_info, const Edge& edge, in GetEdgeShape()
1693 const GraphShapeInfo& shape_info, const Node& node, in GetArgAndRetvalShapes()
4216 const GraphShapeInfo& shape_info, in RewriteTPUReplicateNode()
4470 GraphShapeInfo shape_info; in Run()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.cc62 const Graph& graph, const GraphShapeInfo& graph_shape_info, in ConvertGraphShapeInfoToShapeMap()
458 FunctionLibraryRuntime* flr, GraphShapeInfo* shape_info) { in RunShapeInferenceOnComputation()
513 GraphShapeInfo shape_info; in OptimizeGraph()
526 GraphShapeInfo shape_info; in OptimizeGraph()
Dtpu_compile_op_common.h112 FunctionLibraryRuntime* flr, GraphShapeInfo* shape_info);
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc641 GraphShapeInfo shape_info; in GetGraph()
664 GraphShapeInfo shape_info; in GetGraph()