ker_obj->bag(ker_bag);
ker_obj->layout(coco::KernelLayouts::NHWC::create(as_kernel_shape(ker_shape)));
- // Create a kernel overlay for the kernel object
- // TODO : support for other types
- d->f32()->allocate(ker_obj);
-
- TflBufferContext::TflBuffer<float> buffer = buffer_context.tensor_buffer<float>(graph, ker_idx);
-
- auto ker_spn = d->f32()->weight(ker_bag);
- for (uint32_t idx = 0; idx < buffer.len; ++idx)
- {
- ker_spn[idx] = buffer.ptr[idx];
- }
-
// Create a Load op
auto load = op_builder(m).load(ifm_obj).pop();
bias_obj->bag(bias_bag);
bias_obj->layout(coco::FeatureLayouts::BC::create(ofm_obj->shape()));
- // Fill bias data
- d->f32()->allocate(bias_bag);
-
- auto bias_span = d->f32()->weight(bias_bag);
- buffer = buffer_context.tensor_buffer<float>(graph, bias_idx);
- assert(buffer.ptr != nullptr && buffer.len > 0);
- for (uint32_t idx = 0; idx < buffer.len; ++idx)
- {
- bias_span[idx] = buffer.ptr[idx];
- }
-
// Create Op of conv2d output (last_obj) + bias values(bias_obj)
auto bias_add = op_builder(m).load(last_obj).load(bias_obj).add().pop();
bias_obj->bag(bias_bag);
bias_obj->layout(coco::FeatureLayouts::BC::create(ofm_obj->shape()));
- // Fill bias data
- d->f32()->allocate(bias_bag);
-
- auto bias_span = d->f32()->weight(bias_bag);
- buffer = buffer_context.tensor_buffer<float>(graph, bias_idx);
- assert(buffer.ptr != nullptr && buffer.len > 0);
- for (uint32_t idx = 0; idx < buffer.len; ++idx)
- {
- bias_span[idx] = buffer.ptr[idx];
- }
-
// Create Op of conv2d output (last_obj) + bias values(bias_obj)
auto bias_add = op_builder(m).load(last_obj).load(bias_obj).add().pop();