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
|