Searched refs:device_idx (Results 1 – 3 of 3) sorted by relevance
268 device_idx =[]278 device_idx.append( target_device_idx)279 return (bus_idx, device_idx)
61 int device_idx = step_stats->dev_stats_size(); in MaybeInitializeStepStats() local63 dev_stats->set_device(ctx->devices()->at(device_idx)->name()); in MaybeInitializeStepStats()148 int device_idx = StepStatsDeviceIndex(step_stats, ctx, handle_device); in MaybeCopyInputToExpectedDevice() local149 auto* dev_stats = step_stats->mutable_dev_stats(device_idx); in MaybeCopyInputToExpectedDevice()1000 int device_idx = 0; in EagerKernelExecute() local1003 device_idx = i; in EagerKernelExecute()1008 auto* dev_stats = step_stats->mutable_dev_stats(device_idx); in EagerKernelExecute()
203 void InitializeCollectiveParamsForBroadcast(int instance_key, int device_idx, in InitializeCollectiveParamsForBroadcast() argument215 "/job:localhost/replica:0/task:0/device:CPU:", device_idx)); in InitializeCollectiveParamsForBroadcast()