Searched refs:op_attr (Results 1 – 24 of 24) sorted by relevance
40 const auto& attr = absl::any_cast<const PReLUAttributes&>(ctx.op_attr); in GenerateCode()89 const auto& attr = absl::any_cast<const PReLUAttributes&>(ctx.op_attr); in GenerateCode()156 const auto& attr = absl::any_cast<const PReLUAttributes&>(ctx.op_attr); in GenerateCode()
37 const auto& attr = absl::any_cast<const ConcatAttributes&>(ctx.op_attr); in IsSupported()95 const auto& attr = absl::any_cast<const ConcatAttributes&>(ctx.op_attr); in IsSupported()308 const auto& attr = absl::any_cast<const ConcatAttributes&>(ctx.op_attr); in IsSupported()367 const auto& attr = absl::any_cast<const ConcatAttributes&>(ctx.op_attr); in IsSupported()
44 absl::any_cast<const QuantizeAndDequantizeAttributes&>(ctx.op_attr); in GenerateCode()
37 absl::any_cast<const SpaceToDepthAttributes&>(ctx.op_attr); in GenerateCode()
39 absl::any_cast<const MaxUnpooling2DAttributes&>(ctx.op_attr); in GenerateCode()
38 const auto& attr = absl::any_cast<const ReLUAttributes&>(ctx.op_attr); in GenerateCode()
44 const auto& attr = absl::any_cast<const ReshapeAttributes&>(ctx.op_attr); in GenerateCode()
40 absl::any_cast<const FullyConnectedAttributes&>(ctx.op_attr); in GenerateCode()
45 absl::any_cast<const ConvolutionTransposedAttributes&>(ctx.op_attr); in GenerateCode()
38 const auto& attr = absl::any_cast<const SliceAttributes&>(ctx.op_attr); in GenerateCode()
40 absl::any_cast<const ElementwiseAttributes&>(ctx.op_attr); in GenerateCode()
39 const auto& attr = absl::any_cast<const PadAttributes&>(ctx.op_attr); in GenerateCode()
38 const auto& attr = absl::any_cast<const Resize2DAttributes&>(ctx.op_attr); in GenerateCode()
46 const auto& attr = absl::any_cast<const SoftmaxAttributes&>(ctx.op_attr); in GenerateCode()
47 absl::any_cast<const Convolution2DAttributes&>(ctx.op_attr); in GenerateCode()172 absl::any_cast<const Convolution2DAttributes&>(ctx.op_attr); in GenerateCode()
84 const auto& attr = absl::any_cast<const ElementwiseAttributes&>(ctx.op_attr); in GenerateMultiplyScalarCode()
46 absl::any_cast<const DepthwiseConvolution2DAttributes&>(ctx.op_attr); in GenerateCode()
178 const auto& attr = absl::any_cast<const MeanAttributes&>(ctx.op_attr); in GenerateCode()
178 const auto& attr = absl::any_cast<const Pooling2DAttributes&>(ctx.op_attr); in GenerateCode()
327 for (const auto& op_attr : op_def.attr()) { in InitApiDefFromOpDef() local329 api_attr->set_name(op_attr.name()); in InitApiDefFromOpDef()330 api_attr->set_rename_to(op_attr.name()); in InitApiDefFromOpDef()331 if (op_attr.has_default_value()) { in InitApiDefFromOpDef()332 *api_attr->mutable_default_value() = op_attr.default_value(); in InitApiDefFromOpDef()334 api_attr->set_description(op_attr.description()); in InitApiDefFromOpDef()
104 const absl::any& op_attr; member
149 for (const auto& op_attr : op.attr()) { in TestAllApiDefAttributeNamesAreValid() local150 if (api_def_attr.name() == op_attr.name()) { in TestAllApiDefAttributeNamesAreValid()
1166 struct rtattr *op_attr, in flower_print_arp_op() argument1169 flower_print_masked_u8(f, name, op_attr, mask_attr, in flower_print_arp_op()
346 ArrayAttr:$op_attr