Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Ddirect_session.cc306 run_options.inter_op_thread_pool() > 0) in ShouldUseRunHandlerPool()
583 run_in_caller_thread_ || run_options.inter_op_thread_pool() == -1; in RunInternal()
599 if (run_options.inter_op_thread_pool() < -1 || in RunInternal()
600 run_options.inter_op_thread_pool() >= in RunInternal()
603 run_options.inter_op_thread_pool()); in RunInternal()
606 pool = thread_pools_[run_options.inter_op_thread_pool()].first; in RunInternal()
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dmonitored_session.py1531 options.inter_op_thread_pool = max(options.inter_op_thread_pool,
1532 incoming_options.inter_op_thread_pool)
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.-run-options.pbtxt19 name: "inter_op_thread_pool"
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dconfig.proto747 int32 inter_op_thread_pool = 3; field
/aosp_15_r20/external/tensorflow/tensorflow/python/client/
H A Dsession_test.py161 run_options.inter_op_thread_pool = (
/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.h3249 ::int32_t inter_op_thread_pool() const; in Swap()
7201 inline ::int32_t RunOptions::inter_op_thread_pool() const { in inter_op_thread_pool() function