Home
last modified time | relevance | path

Searched refs:set_function_library_for_shape_inference (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.h147 void set_function_library_for_shape_inference( in set_function_library_for_shape_inference() function
Dshape_refiner_test.cc1395 m.set_function_library_for_shape_inference(&f_lib); in TestSimpleFunctionInference()
1438 m.set_function_library_for_shape_inference(&empty_f_lib); in TEST_F()
1468 m.set_function_library_for_shape_inference(&f_lib); in TEST_F()
1499 m.set_function_library_for_shape_inference(&f_lib); in TEST_F()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_lib.cc222 shape_refiner.set_function_library_for_shape_inference( in FoldConstants()
/external/tensorflow/tensorflow/c/
Dc_api.cc1536 refiner.set_function_library_for_shape_inference(&graph.flib_def()); in TF_Graph()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc713 shape_refiner_->set_function_library_for_shape_inference(&graph_flib_); in AddNodesToShapeRefiner()