Home
last modified time | relevance | path

Searched defs:InferShapes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dshape_inference_testutil.cc29 Status ShapeInferenceTestutil::InferShapes(ShapeInferenceTestOp op, in InferShapes() function in tensorflow::shape_inference::ShapeInferenceTestutil
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference.cc246 Status InferShapes(Graph* graph, const std::map<int, InferredShape>& arg_shapes, in InferShapes() function
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1880 Status InferShapes(const NodeDef& node, NodeContext* c) { in InferShapes() function in tensorflow::grappler::SymbolicShapeRefiner