Searched refs:input_alloc_attrs (Results 1 – 12 of 12) sorted by relevance
282 AllocatorAttributeVec input_alloc_attrs; in Run() local300 params.input_alloc_attrs = &input_alloc_attrs; in Run()389 input_alloc_attrs.clear(); in Run()390 input_alloc_attrs.resize(num_inputs); in Run()407 input_alloc_attrs[j] = input_alloc_attrs_[input_start_index + j]; in Run()
304 AllocatorAttributeVec* input_alloc_attrs,496 saved_input_alloc_attrs(*p.input_alloc_attrs), in AsyncState()506 params.input_alloc_attrs = &saved_input_alloc_attrs; in AsyncState()686 AllocatorAttributeVec input_alloc_attrs; in Process() local711 params.input_alloc_attrs = &input_alloc_attrs; in Process()785 s = PrepareInputs(item, first_input, &inputs, &input_alloc_attrs, in Process()855 AllocatorAttributeVec* input_alloc_attrs, bool* is_input_dead) { in PrepareInputs() argument857 input_alloc_attrs->resize(item.num_inputs); in PrepareInputs()865 (*input_alloc_attrs)[i] = entry->alloc_attr; in PrepareInputs()
89 sub_params_.input_alloc_attrs = &sub_input_attr_;
382 op_params.input_alloc_attrs = &input_aa; in DoPermute()
460 op_params.input_alloc_attrs = &input_aa; in DoGather()
492 op_params.input_alloc_attrs = &input_aa; in DoReduce()
640 op_params.input_alloc_attrs = &input_aa; in DoBroadcast()
674 const gtl::InlinedVector<AllocatorAttributes, 4>* input_alloc_attrs = member1077 if (params_->input_alloc_attrs == nullptr) { in input_alloc_attr()1081 DCHECK_LT(index, params_->input_alloc_attrs->size()); in input_alloc_attr()1082 return (*params_->input_alloc_attrs)[index]; in input_alloc_attr()
553 const auto input_attr = params_->input_alloc_attrs == nullptr in forward_input()
254 params.input_alloc_attrs = &input_alloc_attrs_; in Run()
309 op_params.input_alloc_attrs = &input_aa; in RunReduce()
316 params->input_alloc_attrs = nullptr; in PrepOpKernel()