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

  /external/tensorflow/tensorflow/compiler/xla/service/
backend.h 29 #include "tensorflow/compiler/xla/service/computation_placer.h"
95 ComputationPlacer* computation_placer() const { return computation_placer_; } function in class:xla::Backend
162 ComputationPlacer* computation_placer,
backend.cc 113 TF_ASSIGN_OR_RETURN(auto computation_placer,
117 computation_placer, options.intra_op_parallelism_threads()));
146 ComputationPlacer* computation_placer,
151 computation_placer_(computation_placer) {
local_service.cc 184 return backend().computation_placer()->DeviceId(
hlo_runner.cc 421 backend().computation_placer()->AssignDevices(options.num_replicas, 1));
service.cc 31 #include "tensorflow/compiler/xla/service/computation_placer.h"
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/python/
local_computation_builder.cc 33 #include "tensorflow/compiler/xla/service/computation_placer.h"
357 client.client()->backend().computation_placer()->AssignDevices(

Completed in 260 milliseconds