Searched refs:SpaceToDepthAttributes (Results 1 – 10 of 10) sorted by relevance
38 const SpaceToDepthAttributes attr = {.block_size = 2}; in TEST()52 const SpaceToDepthAttributes attr = {.block_size = 2}; in TEST()68 const SpaceToDepthAttributes attr = {.block_size = 2}; in TEST()87 const SpaceToDepthAttributes attr = {.block_size = 2}; in TEST()
37 absl::any_cast<const SpaceToDepthAttributes&>(ctx.op_attr); in GenerateCode()
33 const SpaceToDepthAttributes attr = {.block_size = 2}; in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()60 const SpaceToDepthAttributes attr = {.block_size = 2}; in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()89 const SpaceToDepthAttributes attr = {.block_size = 2}; in SpaceToDepthTensorShape1x2x2x3BlockSize2Test()120 const SpaceToDepthAttributes attr = {.block_size = 2}; in SpaceToDepthTensorShape1x4x4x1BlockSize2Test()
28 const SpaceToDepthAttributes& attr);
74 const SpaceToDepthAttributes& attr) { in CreateSpaceToDepth()
81 void SelectSpaceToDepth(const SpaceToDepthAttributes& attr,
131 void SelectSpaceToDepth(const SpaceToDepthAttributes& attr, in SelectSpaceToDepth()
490 absl::any_cast<SpaceToDepthAttributes>(node.operation.attributes); in GPUOperationFromNode()
547 struct SpaceToDepthAttributes { struct
1815 SpaceToDepthAttributes attr; in Parse()