Searched refs:PadAttributes (Results 1 – 15 of 15) sorted by relevance
34 PadAttributes attr; in PaddingAppendWidthTest()64 PadAttributes attr; in PaddingPrependWidthTest()94 PadAttributes attr; in PaddingAppendHeightTest()123 PadAttributes attr; in PaddingPrependHeightTest()152 PadAttributes attr; in PaddingAppendChannelsTest()181 PadAttributes attr; in PaddingPrependChannelsTest()210 PadAttributes attr; in PaddingPrependChannelsX4Test()239 PadAttributes attr; in PaddingComplexTest()271 PadAttributes attr; in PaddingReflectWidthTest()301 PadAttributes attr; in PaddingReflectChannelsTest()
28 const PadAttributes& attr, GPUOperation* op) { in GetPaddingCode()139 const PadAttributes& attr) { in CreatePadding()
28 const PadAttributes& attr);
58 PadAttributes pad_attr = in ApplyToNodesSequence()59 absl::any_cast<PadAttributes>(pad_node->operation.attributes); in ApplyToNodesSequence()130 PadAttributes pad_attr = in ApplyToNode()131 absl::any_cast<PadAttributes>(node->operation.attributes); in ApplyToNode()
43 PadAttributes attr; in TEST()80 PadAttributes attr; in TEST()128 PadAttributes pad_attr; in TEST()166 PadAttributes pad_attr; in TEST()
74 auto pad_attr = absl::any_cast<PadAttributes>(pad_node->operation.attributes); in TEST()
67 PadAttributes pad_attr; in ApplyToNode()
48 PadAttributes attr; in TestPadOperation()134 PadAttributes attr; in TEST()157 PadAttributes attr; in TEST()
42 const auto& attr = std::any_cast<const PadAttributes&>(ctx.op_attr); in GenerateCode()
481 struct PadAttributes { struct489 BHWC CalculateOutputShape(const BHWC& input, const PadAttributes& attr); argument
603 BHWC CalculateOutputShape(const BHWC& input, const PadAttributes& attr) { in CalculateOutputShape()
1630 PadAttributes attr; in Parse()
75 void SelectPadding(const PadAttributes& attr, const OperationDef& op_def,
181 void SelectPadding(const PadAttributes& attr, const OperationDef& op_def, in SelectPadding()
609 auto attr = absl::any_cast<PadAttributes>(node.operation.attributes); in GPUOperationFromNodePart0()