Searched refs:input_device_contexts (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor.cc | 199 DeviceContextVec input_device_contexts; in RunAsync() local
222 params.input_device_contexts = &input_device_contexts; in RunAsync()
257 input_device_contexts.clear(); in RunAsync()
258 input_device_contexts.resize(num_inputs); in RunAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 222 gtl::InlinedVector<DeviceContext*, 4> input_device_contexts; in Run() local
228 input_device_contexts.push_back(device_context); in Run()
268 params.input_device_contexts = &input_device_contexts; in Run()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 658 const gtl::InlinedVector<DeviceContext*, 4>* input_device_contexts = member
1619 DCHECK_NE(params_->input_device_contexts, nullptr); in input_device_context()
1621 DCHECK_LT(index, params_->input_device_contexts->size()); in input_device_context()
1624 return static_cast<T*>((*params_->input_device_contexts)[index]); in input_device_context()
1628 DCHECK_NE(params_->input_device_contexts, nullptr); in input_device_context()
1630 DCHECK_LT(index, params_->input_device_contexts->size()); in input_device_context()
1631 return (*params_->input_device_contexts)[index]; in input_device_context()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor.cc | 1321 DeviceContextVec* input_device_contexts,
1549 saved_input_device_contexts(*p.input_device_contexts), in AsyncState()
1560 params.input_device_contexts = &saved_input_device_contexts; in AsyncState()
1619 DeviceContextVec input_device_contexts; in Process() local
1640 params.input_device_contexts = &input_device_contexts; in Process()
1715 s = PrepareInputs(item, first_input, &inputs, &input_device_contexts, in Process()
1880 DeviceContextVec* input_device_contexts, in PrepareInputs() argument
1887 input_device_contexts->clear(); in PrepareInputs()
1888 input_device_contexts->resize(item.num_inputs); in PrepareInputs()
1898 (*input_device_contexts)[i] = entry->device_context; in PrepareInputs()
|
| D | collective_util.cc | 92 sub_params_.input_device_contexts = &sub_input_dc_;
|
| D | ring_gatherer_test.cc | 462 op_params.input_device_contexts = &input_dc; in DoGather()
|
| D | ring_reducer_test.cc | 491 op_params.input_device_contexts = &input_dc; in DoReduce()
|
| D | hierarchical_tree_broadcaster_test.cc | 650 op_params.input_device_contexts = &input_dc; in DoBroadcast()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer_test.cc | 269 op_params.input_device_contexts = &input_dc; in DoReduce()
|