Searched refs:output_alloc_attrs (Results 1 – 4 of 4) sorted by relevance
217 kernel_state.output_alloc_attrs.resize(kernel_state.num_outputs); in Initialize()218 AllocatorAttributes* attrs = kernel_state.output_alloc_attrs.data(); in Initialize()241 kernels_[i].output_alloc_attrs[j]; in Initialize()434 params.output_attr_array = kernel_state.output_alloc_attrs.data(); in Run()514 output_alloc_attrs; // Length = `num_outputs`. member
98 const gtl::InlinedVector<AllocatorAttributes, 1>& output_alloc_attrs() const { in output_alloc_attrs() function
245 params.output_attr_array = runner.output_alloc_attrs().data(); in SetUpParams()
455 std::vector<AllocatorAttributes> output_alloc_attrs({alloc_attrs}); in TEST_F() local456 params.output_attr_array = output_alloc_attrs.data(); in TEST_F()