Searched refs:output_indices_ (Results 1 – 3 of 3) sorted by relevance
169 output_indices_.reserve(output_refs.size()); in Prepare()171 const int64_t object_index = output_indices_.size(); in Prepare()172 output_indices_.push_back(tensor_index); in Prepare()192 for (auto index : output_indices_) { in GetRequiredTemporaries()224 QuantizeOutputs(context, output_indices_, quant_conversion_map_)); in Invoke()235 for (int i = 0; i < output_indices_.size(); ++i) { in SetInputsAndOutputs()237 i, GetTensorObject(output_indices_[i], context))); in SetInputsAndOutputs()349 std::vector<int64_t> output_indices_; member in tflite::gpu::__anon84c75a400111::DelegateKernel
203 output_indices_.reserve(output_refs.size()); in Prepare()205 int object_index = output_indices_.size(); in Prepare()206 output_indices_.push_back(tensor_index); in Prepare()221 for (auto index : output_indices_) { in SetInputsAndOutputs()299 std::vector<int64_t> output_indices_; member in tflite::gpu::cl::__anon7ea745220111::Delegate
158 auto it = output_indices_.find(op); in get_output_index()159 if (it == output_indices_.end()) return llvm::None; in get_output_index()186 output_indices_[alloc_op] = output_index; in find_output_indices()319 DenseMap<Operation *, int32_t> output_indices_; member in mlir::kernel_gen::transforms::__anon8364ced70111::BufferReuseAnalysis