const ::internal::tflite::operand::Index ifm_index{node.param().ifm_index};
const ::internal::tflite::operand::Index paddings_index{node.param().paddings_index};
- const auto ofm_shape = _ctx.at(ofm_index).shape().asFeature();
- const auto ifm_shape = _ctx.at(ifm_index).shape().asFeature();
- const auto paddings_shape = _ctx.at(paddings_index).shape().asTensor();
-
// Set Shape Constraints and TensorInfo
_builder.addShapeConstr(
ifm_index, asTensorInfo(asTensorShape(_ctx.at(ifm_index).shape()), _ctx.at(ifm_index).type(),
const ::internal::tflite::operand::Index ofm_index{node.param().ofm_index};
const ::internal::tflite::operand::Index ifm_index{node.param().ifm_index};
- const auto ofm_shape = _ctx.at(ofm_index).shape();
- const auto ifm_shape = _ctx.at(ifm_index).shape();
-
// Set Shape Constraints and TensorInfo
_builder.addShapeConstr(
ifm_index, asTensorInfo(asTensorShape(_ctx.at(ifm_index).shape()), _ctx.at(ifm_index).type(),
param.ofm_index = ofm_index.asInt();
param.ifm_index = ifm_index.asInt();
- param.radius = 2 * ifm_shape.dim(3) + 1; // normSize = depth * 2 + 1
- param.alpha = 1.0f; // In the implementation to make alpha_ become 1
- param.beta = 0.5f; // pow(reduction, -0.5) = 1 / sqrt(reduction)
- param.bias = 0.0f; // Don't offset the reduction.
+ param.radius = 2 * _ctx.at(ifm_index).shape().dim(3) + 1; // normSize = depth * 2 + 1
+ param.alpha = 1.0f; // In the implementation to make alpha_ become 1
+ param.beta = 0.5f; // pow(reduction, -0.5) = 1 / sqrt(reduction)
+ param.bias = 0.0f; // Don't offset the reduction.
auto stage = [param](const IAllocationContext &ctx, IExecutionBuilder &builder) {
auto ofm_alloc = ctx.at(::internal::tflite::operand::Index{param.ofm_index});