Home
last modified time | relevance | path

Searched defs:device_idx (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc61 int device_idx = step_stats->dev_stats_size(); in MaybeInitializeStepStats() local
148 int device_idx = StepStatsDeviceIndex(step_stats, ctx, handle_device); in MaybeCopyInputToExpectedDevice() local
1000 int device_idx = 0; in EagerKernelExecute() local
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local_test.cc203 void InitializeCollectiveParamsForBroadcast(int instance_key, int device_idx, in InitializeCollectiveParamsForBroadcast()