Home
last modified time | relevance | path

Searched defs:layout_optimizer (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/grappler/
H A Dlayout_optimizer_test.py145 def _get_config(layout_optimizer=True): argument
239 def _train(self, checkpoint_path, layout_optimizer=False, restore=False): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Drewriter_config.proto69 Toggle layout_optimizer = 1; field
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/grappler/
H A Dgrappler.h114 TF_TriState layout_optimizer; member
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Drewriter_config.pb.h1603 inline ::tensorflow::RewriterConfig_Toggle RewriterConfig::layout_optimizer() const { in layout_optimizer() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc5863 tensorflow::grappler::GenericLayoutOptimizer layout_optimizer("NCHW"); in ConvertGraphDefToEngine() local