Searched refs:InferredShape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference.h | 29 struct InferredShape { struct
38 typedef std::unordered_map<string, std::vector<InferredShape>> GraphShapeInfo; argument
43 Status InferShapes(Graph* graph, const std::map<int, InferredShape>& arg_shapes,
49 xla::StatusOr<InferredShape> MergeInferredShapes(const InferredShape& a,
50 const InferredShape& b);
|
| D | shape_inference.cc | 43 const std::map<int, InferredShape>& arg_shapes, in PropagateShapes()
72 const InferredShape& arg_shape = it->second; in PropagateShapes()
135 Status InferShapes(Graph* graph, const std::map<int, InferredShape>& arg_shapes, in InferShapes()
161 xla::StatusOr<InferredShape> MergeInferredShapes(const InferredShape& a, in MergeInferredShapes()
162 const InferredShape& b) { in MergeInferredShapes()
163 InferredShape result; in MergeInferredShapes()
|
| D | encapsulate_util.cc | 315 std::map<int, InferredShape> arg_shapes; in PerformStaticShapeInferenceBeforeEncapsulation()
326 [](const InferredShape& inferred_shape) { in PerformStaticShapeInferenceBeforeEncapsulation()
|