Searched refs:output_alloc_attrs (Results 1 – 6 of 6) sorted by relevance
223 kernel_state.output_alloc_attrs.resize(kernel_state.num_outputs); in Initialize()224 AllocatorAttributes* attrs = kernel_state.output_alloc_attrs.data(); in Initialize()247 kernels_[i].output_alloc_attrs[j]; in Initialize()441 params.output_attr_array = kernel_state.output_alloc_attrs.data(); in Run()529 output_alloc_attrs; // Length = `num_outputs`. member
37 params.output_attr_array = runner.output_alloc_attrs().data(); in SetUpParams()
98 const gtl::InlinedVector<AllocatorAttributes, 1>& output_alloc_attrs() const { in output_alloc_attrs() function
79 const gtl::InlinedVector<AllocatorAttributes, 1>& output_alloc_attrs() const { in output_alloc_attrs() function
458 std::vector<AllocatorAttributes> output_alloc_attrs({alloc_attrs}); in TEST_F() local459 params.output_attr_array = output_alloc_attrs.data(); in TEST_F()
422 op_kernel_runner.output_alloc_attrs().data(); in ExecuteOpKernelRunner()