Home
last modified time | relevance | path

Searched refs:ShapeHandleToProto (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/c/
Dpython_api.cc138 ic->ShapeHandleToProto(p.shape, out_shape_and_type->mutable_shape()); in GetHandleShapeAndType()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.h373 void ShapeHandleToProto(ShapeHandle handle, TensorShapeProto* proto);
Dshape_inference_test.cc985 c.ShapeHandleToProto(s, &proto); in TEST_F()
999 c.ShapeHandleToProto(u0, &proto); in TEST_F()
Dshape_inference.cc285 void InferenceContext::ShapeHandleToProto(ShapeHandle handle, in ShapeHandleToProto() function in tensorflow::shape_inference::InferenceContext
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc109 node_context->ShapeHandleToProto(node_context->input(0), &proto); in InferShapesForFunctionSubNode()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc1989 context->ShapeHandleToProto(shape, &proto); in DoStaticShapeInferenceForOutsideCompilationSend()
2008 context->ShapeHandleToProto(shape, &proto); in DoStaticShapeInferenceForOutsideCompilationSend()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc645 input_ic->ShapeHandleToProto(handle, &proto); in UpdateFunction()