Home
last modified time | relevance | path

Searched refs:input_dev_ptrs (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc322 std::vector<Device*>* input_dev_ptrs, in AddInputDevicesToCacheKey() argument
324 input_dev_ptrs->reserve(op->Inputs().size()); in AddInputDevicesToCacheKey()
339 input_dev_ptrs->push_back(input_device); in AddInputDevicesToCacheKey()
341 input_dev_ptrs->push_back(cpu_device); in AddInputDevicesToCacheKey()
345 input_dev_ptrs->push_back(device == nullptr ? cpu_device : device); in AddInputDevicesToCacheKey()
383 std::vector<Device*> input_dev_ptrs; in EagerLocalExecute() local
386 AddInputDevicesToCacheKey(ctx, op, &input_dev_ptrs, &cache_key)); in EagerLocalExecute()
445 flr, ctx->pflr(), std::move(input_dev_ptrs), runner, in EagerLocalExecute()