Searched refs:fb_op (Results 1 – 2 of 2) sorted by relevance
806 absl::Status Decode(const data::GPUOperation* fb_op, GPUOperation* op) { in Decode() argument807 RETURN_IF_ERROR(Decode(fb_op->arguments(), &op->args_)); in Decode()808 op->code_ = std::string(fb_op->code()->c_str(), fb_op->code()->size()); in Decode()809 op->work_group_size_.x = fb_op->work_group_size()->x(); in Decode()810 op->work_group_size_.y = fb_op->work_group_size()->y(); in Decode()811 op->work_group_size_.z = fb_op->work_group_size()->z(); in Decode()812 for (auto option_fb : *fb_op->compiler_options()) { in Decode()815 op->tensor_to_grid_ = ToEnum(fb_op->tensor_to_grid()); in Decode()816 op->elementwise_ = fb_op->elementwise(); in Decode()817 op->linkable_ = fb_op->linkable(); in Decode()[all …]
159 friend absl::Status Decode(const tflite::gpu::data::GPUOperation* fb_op,