Searched refs:input_device_contexts (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor.cc | 1239 DeviceContextVec* input_device_contexts,
1456 saved_input_device_contexts(*p.input_device_contexts), in AsyncState()
1467 params.input_device_contexts = &saved_input_device_contexts; in AsyncState()
1498 DeviceContextVec input_device_contexts; in Process() local
1517 params.input_device_contexts = &input_device_contexts; in Process()
1579 s = PrepareInputs(item, first_input, &inputs, &input_device_contexts, in Process()
1707 DeviceContextVec* input_device_contexts, in PrepareInputs() argument
1714 input_device_contexts->clear(); in PrepareInputs()
1715 input_device_contexts->resize(item.num_inputs); in PrepareInputs()
1725 (*input_device_contexts)[i] = entry->device_context; in PrepareInputs()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 553 const gtl::InlinedVector<DeviceContext*, 4>* input_device_contexts = member
1424 DCHECK_LT(index, params_->input_device_contexts->size()); in input_device_context()
1427 return static_cast<T*>((*params_->input_device_contexts)[index]); in input_device_context()
1432 DCHECK_LT(index, params_->input_device_contexts->size()); in input_device_context()
1433 return (*params_->input_device_contexts)[index]; in input_device_context()
|