Home
last modified time | relevance | path

Searched refs:add_session_inter_op_thread_pool (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsdca_ops_test.cc34 result->config.add_session_inter_op_thread_pool()->set_num_threads(1); in GetSingleThreadedOptions()
45 result->config.add_session_inter_op_thread_pool()->set_num_threads( in GetMultiThreadedOptions()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/clusters/
H A Dsingle_machine.cc56 options_.config.add_session_inter_op_thread_pool()->set_num_threads( in SingleMachine()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Ddirect_session_test.cc1998 auto* p = options.config.add_session_inter_op_thread_pool(); in TestSessionInterOpThreadsImpl()
2000 p = options.config.add_session_inter_op_thread_pool(); in TestSessionInterOpThreadsImpl()
2140 options.config.add_session_inter_op_thread_pool(); in TEST()
/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/
Dconfig.pb.h2534 ::tensorflow::ThreadPoolOptionProto* add_session_inter_op_thread_pool(); in Swap()
6371 inline ::tensorflow::ThreadPoolOptionProto* ConfigProto::add_session_inter_op_thread_pool() { in add_session_inter_op_thread_pool() function