Searched refs:shape_info (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference_test.cc | 49 GraphShapeInfo shape_info; in TEST() local
51 /*fnlib_def=*/nullptr, &shape_info)); in TEST()
59 TF_EXPECT_OK(ShapeAnnotationsMatch(*graph, shape_info, expected)); in TEST()
113 GraphShapeInfo shape_info; in TEST() local
115 &shape_info)); in TEST()
120 TF_EXPECT_OK(ShapeAnnotationsMatch(graph, shape_info, expected)); in TEST()
|
| D | test_util.cc | 24 const Graph& graph, const GraphShapeInfo& shape_info, in ShapeAnnotationsMatch() argument
27 auto sit = shape_info.find(node->name()); in ShapeAnnotationsMatch()
28 TF_RET_CHECK(sit != shape_info.end()) in ShapeAnnotationsMatch()
|
| D | shape_inference.cc | 99 GraphShapeInfo* shape_info) { in StoreOutputShapes() argument
104 auto& outputs = (*shape_info)[node->name()]; in StoreOutputShapes()
137 GraphShapeInfo* shape_info) { in InferShapes() argument
158 return StoreOutputShapes(*graph, shape_refiner, shape_info); in InferShapes()
|
| D | test_util.h | 38 const Graph& graph, const GraphShapeInfo& shape_info,
|
| D | encapsulate_util.cc | 316 GraphShapeInfo shape_info; in PerformStaticShapeInferenceBeforeEncapsulation() local
318 InferShapes(g, arg_shapes, /*fnlib_def=*/nullptr, &shape_info)); in PerformStaticShapeInferenceBeforeEncapsulation()
322 for (auto iter : shape_info) { in PerformStaticShapeInferenceBeforeEncapsulation()
|
| D | shape_inference.h | 45 GraphShapeInfo* shape_info);
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 631 def _wrap_shape(shape_info): argument
632 dtype, dims = shape_info
641 def _wrap_program_shape(shape_info): argument
642 arg_shapes, result_shape = shape_info
|