Searched refs:PadAttributes (Results 1 – 15 of 15) sorted by relevance
33 PadAttributes attr; in PaddingAppendWidthTest()63 PadAttributes attr; in PaddingPrependWidthTest()93 PadAttributes attr; in PaddingAppendHeightTest()122 PadAttributes attr; in PaddingPrependHeightTest()151 PadAttributes attr; in PaddingAppendChannelsTest()180 PadAttributes attr; in PaddingPrependChannelsTest()209 PadAttributes attr; in PaddingPrependChannelsX4Test()238 PadAttributes attr; in PaddingComplexTest()270 PadAttributes attr; in PaddingReflectWidthTest()300 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()
47 PadAttributes attr; in TestPadOperation()133 PadAttributes attr; in TEST()156 PadAttributes attr; in TEST()
39 const auto& attr = absl::any_cast<const PadAttributes&>(ctx.op_attr); in GenerateCode()
443 struct PadAttributes { struct451 BHWC CalculateOutputShape(const BHWC& input, const PadAttributes& attr); argument
564 BHWC CalculateOutputShape(const BHWC& input, const PadAttributes& attr) { in CalculateOutputShape()
1294 PadAttributes attr; in Parse()
66 void SelectPadding(const PadAttributes& attr, const OperationDef& op_def,
149 void SelectPadding(const PadAttributes& attr, const OperationDef& op_def, in SelectPadding()
443 auto attr = absl::any_cast<PadAttributes>(node.operation.attributes); in GPUOperationFromNode()