Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dshape_refiner.h147 void set_function_library_for_shape_inference( in set_function_library_for_shape_inference() function
H A Dshape_refiner_test.cc1443 m.set_function_library_for_shape_inference(&f_lib); in TestSimpleFunctionInference()
1486 m.set_function_library_for_shape_inference(&empty_f_lib); in TEST_F()
1516 m.set_function_library_for_shape_inference(&f_lib); in TEST_F()
1547 m.set_function_library_for_shape_inference(&f_lib); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dfold_constants_lib.cc222 shape_refiner.set_function_library_for_shape_inference( in FoldConstants()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.cc663 shape_refiner.set_function_library_for_shape_inference(&flib_def); in PrepareGraphForMlir()
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api.cc1561 refiner.set_function_library_for_shape_inference(&graph.flib_def()); in TF_Graph()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dimport_model.cc822 shape_refiner_->set_function_library_for_shape_inference(&graph_flib_); in AddNodesToShapeRefiner()