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