Searched refs:kernel_state (Results 1 – 2 of 2) sorted by relevance
38 for (const KernelState& kernel_state : kernels_) { in ~SingleThreadedExecutorImpl() local39 params_.delete_kernel(kernel_state.kernel); in ~SingleThreadedExecutorImpl()92 KernelState& kernel_state = kernels_[i]; in Initialize() local93 TF_RETURN_IF_ERROR(params_.create_kernel(n->def(), &kernel_state.kernel)); in Initialize()94 kernel_state.num_inputs = n->num_inputs(); in Initialize()95 kernel_state.num_outputs = n->num_outputs(); in Initialize()98 kernel_state.input_start_index = 0; in Initialize()101 kernel_state.input_start_index = in Initialize()111 KernelState& kernel_state = kernels_[i]; in Initialize() local112 kernel_state.output_locations.resize(kernel_state.num_outputs); in Initialize()[all …]
1334 NNAPIDelegateKernel* kernel_state = new NNAPIDelegateKernel; in __anon2e5b0dd81702() local1335 kernel_state->Init(context, params); in __anon2e5b0dd81702()1336 return kernel_state; in __anon2e5b0dd81702()