Home
last modified time | relevance | path

Searched refs:PropagateShapes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference.cc42 Status PropagateShapes(const Graph& graph, in PropagateShapes() function
152 TF_RETURN_IF_ERROR(PropagateShapes(*graph, arg_shapes, &shape_refiner)); in InferShapes()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.h166 Status PropagateShapes(
Dgraph_properties.cc1892 Status GraphProperties::PropagateShapes( in PropagateShapes() function in tensorflow::grappler::GraphProperties
2138 PropagateShapes(&refiner, &new_shapes, resource_handles, num_loops)); in InferStatically()