Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dinterpreter_options.h65 void OptimizeMemoryForLargeTensors(int value = 1 << 20) {
H A Dinterpreter.cc465 subgraph->OptimizeMemoryForLargeTensors( in ApplyOptionsImpl()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dwhile_test.cc83 options.OptimizeMemoryForLargeTensors(1000000); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/
H A Dsubgraph.h423 void OptimizeMemoryForLargeTensors(int large_tensors_thresholds_in_bytes);
H A Dsubgraph.cc1607 void Subgraph::OptimizeMemoryForLargeTensors( in OptimizeMemoryForLargeTensors() function in tflite::Subgraph
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/benchmark/
H A Dbenchmark_tflite_model.cc720 options.OptimizeMemoryForLargeTensors( in Init()