Searched refs:InferShapes (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/core/framework/ |
D | shape_inference_testutil.h | 69 static Status InferShapes(ShapeInferenceTestOp op, const string& ins, 84 "", ::tensorflow::shape_inference::ShapeInferenceTestutil::InferShapes( \ 90 ::tensorflow::shape_inference::ShapeInferenceTestutil::InferShapes( \
|
D | shape_inference_testutil_test.cc | 64 return ShapeInferenceTestutil::InferShapes(op, ins, expected_outs) in RunInferShapes() 99 auto error_message = ShapeInferenceTestutil::InferShapes( in TEST()
|
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_test.cc | 50 TF_ASSERT_OK(InferShapes(graph.get(), /*arg_shapes=*/{}, in TEST() 114 TF_ASSERT_OK(InferShapes(&graph, /*arg_shapes=*/{}, /*fnlib_def=*/nullptr, in TEST()
|
D | shape_inference.h | 43 Status InferShapes(Graph* graph, const std::map<int, InferredShape>& arg_shapes,
|
D | shape_inference.cc | 135 Status InferShapes(Graph* graph, const std::map<int, InferredShape>& arg_shapes, in InferShapes() function
|
D | encapsulate_util.cc | 318 InferShapes(g, arg_shapes, /*fnlib_def=*/nullptr, &shape_info)); in PerformStaticShapeInferenceBeforeEncapsulation()
|
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | graph_properties.cc | 885 return InferShapes(*node, ctx); in UpdateNode() 1580 Status InferShapes(const NodeDef& node, NodeContext* c) { in InferShapes() function in tensorflow::grappler::SymbolicShapeRefiner
|