Home
last modified time | relevance | path

Searched defs:num_computations (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/ops/
Dtpu_compile_op.cc36 int num_computations; in __anondb58f4d00102() local
96 int num_computations; in __anondb58f4d00302() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_impl.h40 int num_computations, bool return_hlo_protos, in TpuCompileOpKernelImpl()
48 int num_computations, bool return_hlo_protos, in TpuCompileOpKernelImpl()
Dtpu_compile_op_common.h70 int num_computations, bool return_hlo_protos, in TpuCompileOpKernelCommon()
84 int num_computations, bool return_hlo_protos, in TpuCompileOpKernelCommon()
Dtpu_compile_op_support.cc393 int num_computations; in CompileOpMetadataFromContext() local
/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_executable.cc143 int num_computations = executable_->addressable_devices().size(); in ExecuteShardedOnLocalDevices() local
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc759 int num_computations = args.front().size(); in ExecuteShardedOnLocalDevices() local