Searched defs:num_computations (Results 1 – 6 of 6) sorted by relevance
36 int num_computations; in __anon78c361350102() local95 int num_computations; in __anon78c361350302() local
40 int num_computations, bool return_hlo_protos, in TpuCompileOpKernelImpl()48 int num_computations, bool return_hlo_protos, in TpuCompileOpKernelImpl()
68 int num_computations, bool return_hlo_protos, in TpuCompileOpKernelCommon()82 int num_computations, bool return_hlo_protos, in TpuCompileOpKernelCommon()
399 int num_computations; in CompileOpMetadataFromContext() local
95 int num_computations = executable_->addressable_devices().size(); in ExecuteShardedOnLocalDevices() local
759 int num_computations = args.front().size(); in ExecuteShardedOnLocalDevices() local