Home
last modified time | relevance | path

Searched refs:xrt_computation_proto (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_program_group.cc281 const xrt::XLAComputation& xrt_computation_proto, in CompileAndBuild() argument
285 se_tpu::SerializeProto(xrt_computation_proto); in CompileAndBuild()
302 xrt_computation_proto.config().num_cores_per_replica() in CompileAndBuild()
303 ? xrt_computation_proto.config().num_cores_per_replica() in CompileAndBuild()
Dtpu_program_group.h100 const xrt::XLAComputation& xrt_computation_proto,