Searched refs:GraphShapeInfo (Results 1 – 11 of 11) sorted by relevance
/external/tensorflow/tensorflow/compiler/jit/ |
D | shape_inference.h | 38 typedef std::unordered_map<string, std::vector<InferredShape>> GraphShapeInfo; typedef 45 GraphShapeInfo* shape_info);
|
D | shape_inference_test.cc | 51 GraphShapeInfo shape_info; in TEST() 115 GraphShapeInfo shape_info; in TEST() 161 GraphShapeInfo shape_info; in TEST()
|
D | test_util.h | 41 const Graph& graph, const GraphShapeInfo& shape_info,
|
D | test_util.cc | 24 const Graph& graph, const GraphShapeInfo& shape_info, in ShapeAnnotationsMatch()
|
D | shape_inference.cc | 210 GraphShapeInfo* shape_info) { in StoreOutputShapes() 248 GraphShapeInfo* shape_info) { in InferShapes()
|
D | encapsulate_util.cc | 326 GraphShapeInfo shape_info; in PerformStaticShapeInferenceBeforeEncapsulation()
|
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
D | distributed_tpu_rewrite_pass.h | 294 const GraphShapeInfo& shape_info, const Node& node, 557 const GraphShapeInfo& shape_info,
|
D | distributed_tpu_rewrite_pass.cc | 1679 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/ |
D | tpu_compile_op_common.cc | 62 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()
|
D | tpu_compile_op_common.h | 112 FunctionLibraryRuntime* flr, GraphShapeInfo* shape_info);
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_compiler.cc | 641 GraphShapeInfo shape_info; in GetGraph() 664 GraphShapeInfo shape_info; in GetGraph()
|