Searched refs:input_alloc_attrs (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | single_threaded_executor.cc | 290 AllocatorAttributeVec input_alloc_attrs; in Run() local
319 params.input_alloc_attrs = &input_alloc_attrs; in Run()
413 input_alloc_attrs.clear(); in Run()
414 input_alloc_attrs.resize(num_inputs); in Run()
431 input_alloc_attrs[j] = input_alloc_attrs_[input_start_index + j]; in Run()
|
| D | executor.cc | 304 AllocatorAttributeVec* input_alloc_attrs,
501 saved_input_alloc_attrs(*p.input_alloc_attrs), in AsyncState()
511 params.input_alloc_attrs = &saved_input_alloc_attrs; in AsyncState()
691 AllocatorAttributeVec input_alloc_attrs; in Process() local
718 params.input_alloc_attrs = &input_alloc_attrs; in Process()
792 s = PrepareInputs(item, first_input, &inputs, &input_alloc_attrs, in Process()
862 AllocatorAttributeVec* input_alloc_attrs, bool* is_input_dead) { in PrepareInputs() argument
864 input_alloc_attrs->resize(item.num_inputs); in PrepareInputs()
872 (*input_alloc_attrs)[i] = entry->alloc_attr; in PrepareInputs()
|
| D | collective_util.cc | 89 sub_params_.input_alloc_attrs = &sub_input_attr_;
|
| D | collective_test_util.cc | 316 op_params.input_alloc_attrs = &input_aa; in RunCollective()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | op_kernel_runner.h | 95 const gtl::InlinedVector<AllocatorAttributes, 4>& input_alloc_attrs() const { in input_alloc_attrs() function
|
| D | kernel_fallback_execute_compat.cc | 244 params.input_alloc_attrs = &runner.input_alloc_attrs(); in SetUpParams()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 634 const gtl::InlinedVector<AllocatorAttributes, 4>* input_alloc_attrs = member
1022 if (params_->input_alloc_attrs == nullptr) { in input_alloc_attr()
1026 DCHECK_LT(index, params_->input_alloc_attrs->size()); in input_alloc_attr()
1027 return (*params_->input_alloc_attrs)[index]; in input_alloc_attr()
|
| D | op_kernel.cc | 530 const auto input_attr = params_->input_alloc_attrs == nullptr in forward_input()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 261 params.input_alloc_attrs = &input_alloc_attrs_; in Run()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 312 op_params.input_alloc_attrs = &input_aa; in RunReduce()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 316 params->input_alloc_attrs = nullptr; in PrepOpKernel()
|