Searched defs:InferShapes (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ | ||
| H A D | shape_inference_testutil.cc | 29 Status ShapeInferenceTestutil::InferShapes(ShapeInferenceTestOp op, in InferShapes() function in tensorflow::shape_inference::ShapeInferenceTestutil |
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ | ||
| H A D | shape_inference.cc | 243 Status InferShapes(Graph* graph, const std::map<int, InferredShape>& arg_shapes, in InferShapes() function |
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ | ||
| H A D | graph_properties.cc | 1919 Status InferShapes(const NodeDef& node, NodeContext* c) { in InferShapes() function in tensorflow::grappler::SymbolicShapeRefiner |