Searched defs:num_computations (Results 1 – 6 of 6) sorted by relevance
36 int num_computations; in __anondb58f4d00102() local96 int num_computations; in __anondb58f4d00302() local
40 int num_computations, bool return_hlo_protos, in TpuCompileOpKernelImpl()48 int num_computations, bool return_hlo_protos, in TpuCompileOpKernelImpl()
70 int num_computations, bool return_hlo_protos, in TpuCompileOpKernelCommon()84 int num_computations, bool return_hlo_protos, in TpuCompileOpKernelCommon()
393 int num_computations; in CompileOpMetadataFromContext() local
143 int num_computations = executable_->addressable_devices().size(); in ExecuteShardedOnLocalDevices() local
759 int num_computations = args.front().size(); in ExecuteShardedOnLocalDevices() local