Searched refs:input_alloc_attrs (Results 1 – 10 of 10) sorted by relevance
200 AllocatorAttributeVec input_alloc_attrs; in RunAsync() local223 params.input_alloc_attrs = &input_alloc_attrs; in RunAsync()249 input_alloc_attrs.clear(); in RunAsync()250 input_alloc_attrs.resize(num_inputs); in RunAsync()254 input_alloc_attrs[j] = input_alloc_attrs_[input_start_index + j]; in RunAsync()
1322 AllocatorAttributeVec* input_alloc_attrs,1550 saved_input_alloc_attrs(*p.input_alloc_attrs), in AsyncState()1561 params.input_alloc_attrs = &saved_input_alloc_attrs; in AsyncState()1620 AllocatorAttributeVec input_alloc_attrs; in Process() local1641 params.input_alloc_attrs = &input_alloc_attrs; in Process()1716 &input_alloc_attrs, &is_input_dead); in Process()1881 AllocatorAttributeVec* input_alloc_attrs, in PrepareInputs() argument1889 input_alloc_attrs->clear(); in PrepareInputs()1890 input_alloc_attrs->resize(item.num_inputs); in PrepareInputs()1899 (*input_alloc_attrs)[i] = entry->alloc_attr; in PrepareInputs()
91 sub_params_.input_alloc_attrs = &sub_input_attr_;
451 op_params.input_alloc_attrs = &input_aa; in DoGather()
480 op_params.input_alloc_attrs = &input_aa; in DoReduce()
639 op_params.input_alloc_attrs = &input_aa; in DoBroadcast()
654 const gtl::InlinedVector<AllocatorAttributes, 4>* input_alloc_attrs = member1042 if (params_->input_alloc_attrs == nullptr) { in input_alloc_attr()1046 DCHECK_LT(index, params_->input_alloc_attrs->size()); in input_alloc_attr()1047 return (*params_->input_alloc_attrs)[index]; in input_alloc_attr()
551 const auto input_attr = params_->input_alloc_attrs == nullptr in forward_input()
258 op_params.input_alloc_attrs = &input_aa; in DoReduce()
237 params.input_alloc_attrs = &in_attrs; in Run()