Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dxla_fusion_optimizer.cc140 shape_refiner.set_require_shape_inference_fns(false); in Optimize()
Dshape_inference.cc139 shape_refiner.set_require_shape_inference_fns(false); in InferShapes()
Dencapsulate_subgraphs_pass.cc2373 shape_refiner->set_require_shape_inference_fns(false); in MakeGraphForOutsideCompilationSends()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.h150 void set_require_shape_inference_fns(bool require_shape_inference_fns) { in set_require_shape_inference_fns() function
Dshape_refiner_test.cc233 m.set_require_shape_inference_fns(false); in TEST_F()
1272 m.set_require_shape_inference_fns(false); in TEST_F()
1299 m.set_require_shape_inference_fns(false); in TEST_F()
/external/tensorflow/tensorflow/c/
Dpython_api.cc114 graph->refiner.set_require_shape_inference_fns(require); in SetRequireShapeInferenceFns()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_lib.cc220 shape_refiner.set_require_shape_inference_fns(false); in FoldConstants()