Searched refs:threading_options (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | finalize_dataset_op.cc | 53 ctx, input, options.threading_options().max_intra_op_parallelism(), in MakeDatasetHelper()
60 ctx, input, options.threading_options().private_threadpool_size(), in MakeDatasetHelper()
|
| /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/framework/ |
| D | dataset_options.pb.h | 1535 const ::tensorflow::data::ThreadingOptions& threading_options() const; in Swap()
1538 void set_allocated_threading_options(::tensorflow::data::ThreadingOptions* threading_options); in Swap()
1544 ::tensorflow::data::ThreadingOptions* threading_options); in Swap()
2922 inline const ::tensorflow::data::ThreadingOptions& Options::threading_options() const { in threading_options() function
2927 ::tensorflow::data::ThreadingOptions* threading_options) { in unsafe_arena_set_allocated_threading_options() argument
2931 _impl_.threading_options_ = threading_options; in unsafe_arena_set_allocated_threading_options()
2969 …Options::set_allocated_threading_options(::tensorflow::data::ThreadingOptions* threading_options) { in set_allocated_threading_options() argument
2974 if (threading_options) { in set_allocated_threading_options()
2976 ::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(threading_options); in set_allocated_threading_options()
2978 threading_options = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage( in set_allocated_threading_options()
[all …]
|
| D | dataset_options.pb.cc | 2356 static const ::tensorflow::data::ThreadingOptions& threading_options(const Options* msg);
2372 Options::_Internal::threading_options(const Options* msg) { in threading_options() function in tensorflow::data::Options::_Internal
2690 InternalWriteMessage(5, _Internal::threading_options(this), in _InternalSerialize()
2691 _Internal::threading_options(this).GetCachedSize(), target, stream); in _InternalSerialize()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | root_dataset.cc | 57 options.threading_options().max_intra_op_parallelism(); in SetRootDatasetParams()
61 options.threading_options().private_threadpool_size(); in SetRootDatasetParams()
|
| H A D | dataset_utils.cc | 861 return options.threading_options().optional_max_intra_op_parallelism_case() == in ShouldConfigureMaxIntraOpParallelism()
866 return options.threading_options().optional_private_threadpool_size_case() == in ShouldUsePrivateThreadPool()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/ |
| H A D | options.py | 625 pb.threading_options.CopyFrom(self.threading._to_proto()) # pylint: disable=protected-access
640 self.threading._from_proto(pb.threading_options) # pylint: disable=protected-access
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | dataset_options.proto | 196 ThreadingOptions threading_options = 5; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/ |
| H A D | options_test.py | 186 expected_pb.threading_options.CopyFrom(
|