HomeSort by relevance Sort by last modified time
    Searched refs:device_assignment_ (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/
executable_run_options.cc 81 device_assignment_ = device_assignment;
86 return device_assignment_;
executable_run_options.h 86 const DeviceAssignment* device_assignment_ = nullptr; member in class:xla::ExecutableRunOptions
  /external/tensorflow/tensorflow/compiler/xla/python/
local_computation_builder.cc 214 device_assignment_(std::move(device_assignment)),
218 int num_replicas = device_assignment_.replica_count();
222 device_ordinals.push_back(device_assignment_(i, 0));
235 const int device_ordinal = device_assignment_(0, 0);
250 options.set_device_assignment(&device_assignment_);
283 const int device_ordinal = device_assignment_(replica, 0);
298 options.set_device_assignment(&device_assignment_);
local_computation_builder.h 148 const xla::DeviceAssignment device_assignment_; member in class:xla::swig::LocalExecutable

Completed in 69 milliseconds