HomeSort by relevance Sort by last modified time
    Searched refs:device_count (Results 51 - 75 of 81) sorted by null

1 23 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 23 4