OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:device_count
(Results
51 - 75
of
81
) sorted by null
1
2
3
4
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
unidirectional_sequence_lstm_test.py
32
CONFIG = tf.ConfigProto(
device_count
={"GPU": 0})
unidirectional_sequence_rnn_test.py
35
CONFIG = tf.ConfigProto(
device_count
={"GPU": 0})
bidirectional_sequence_rnn_test.py
37
CONFIG = tf.ConfigProto(
device_count
={"GPU": 0})
/external/tensorflow/tensorflow/python/kernel_tests/
functional_ops_test.py
409
worker_config.
device_count
["CPU"] = 2
436
worker_config.
device_count
["CPU"] = 2
[
all
...]
cond_v2_test.py
[
all
...]
/external/tensorflow/tensorflow/stream_executor/cuda/
cuda_driver.cc
1265
int
device_count
= 0;
local
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/python/
xla_client.py
80
def
device_count
(self):
member in class:Backend
129
def
device_count
(self):
member in class:XlaLocalBackend
166
def
device_count
(self):
member in class:XrtBackend
[
all
...]
local_computation_builder.cc
81
int LocalClient::DeviceCount() const { return client_->
device_count
(); }
266
const int num_devices = client_->
device_count
();
/external/tensorflow/tensorflow/lite/delegates/nnapi/
nnapi_delegate.cc
[
all
...]
/external/tensorflow/tensorflow/contrib/distribute/python/
multi_worker_test_base.py
154
ps_config.
device_count
['GPU'] = 0
/external/tensorflow/tensorflow/python/data/kernel_tests/
from_tensors_test.py
259
config=config_pb2.ConfigProto(
device_count
={"CPU": 2})) as sess:
map_test.py
[
all
...]
iterator_test.py
599
worker_config.
device_count
["CPU"] = 3
[
all
...]
/external/tensorflow/tensorflow/python/eager/
function_argument_naming_test.py
257
config=config_pb2.ConfigProto(
device_count
={'CPU': 4}))
core_test.py
193
device_count
={'GPU': 0}))
function_gradients_test.py
757
config=config_pb2.ConfigProto(
device_count
={'CPU': 4}))
/external/tensorflow/tensorflow/core/common_runtime/
process_function_library_runtime_test.cc
70
auto*
device_count
= options.config.mutable_device_count();
local
71
device_count
->insert({"CPU", 2});
function_test.cc
149
auto*
device_count
= options.config.mutable_device_count();
local
150
device_count
->insert({"CPU", 3});
[
all
...]
/external/tensorflow/tensorflow/python/saved_model/
saved_model_test.py
[
all
...]
/external/tensorflow/tensorflow/contrib/training/python/training/
evaluation_test.py
98
config=config_pb2.ConfigProto(
device_count
={'CPU': 2})) as session:
/external/libdrm/
xf86drm.c
3960
int ret, i, node_count,
device_count
;
local
[
all
...]
/external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_device.cc
[
all
...]
/external/tensorflow/tensorflow/python/client/
session_clusterspec_prop_test.py
183
server_config = config_pb2.ConfigProto(
device_count
={'CPU': 2})
/external/tensorflow/tensorflow/python/training/
saver_test.py
[
all
...]
/external/tensorflow/tensorflow/compiler/jit/
encapsulate_subgraphs_pass.cc
2544
auto*
device_count
= session_options.config.mutable_
device_count
();
local
[
all
...]
Completed in 1333 milliseconds
1
2
3
4