Searched refs:ConcatAttributes (Results 1 – 13 of 13) sorted by relevance
47 ConcatAttributes attr; in TEST()73 ConcatAttributes attr; in TEST()99 ConcatAttributes attr; in TEST()125 ConcatAttributes attr; in TEST()
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()
37 ConcatAttributes attr; in ConcatWidthTest()70 ConcatAttributes attr; in ConcatHeightTest()104 ConcatAttributes attr; in ConcatChannelsTest()142 ConcatAttributes attr; in ConcatChannelsAlignedx4Test()
30 const ConcatAttributes& attr) { in GetConcatKernelCode()115 const ConcatAttributes& attr) { in CreateConcatXY()
28 const ConcatAttributes& attr);
198 struct ConcatAttributes { struct224 const ConcatAttributes& attr,230 const ConcatAttributes& attr,
602 const ConcatAttributes& attr, in CalculateOutputShape()659 const ConcatAttributes& attr, in CalculateOutputShape()
247 ConcatAttributes attr; in Parse()962 ConcatAttributes concat_attr; in ParseBasic()1196 ConcatAttributes attr; in Parse()
43 ConcatAttributes attr; in TEST()
66 absl::any_cast<const ConcatAttributes&>(node->operation.attributes); in ApplyToNode()
53 absl::Status SelectConcat(const ConcatAttributes& attr,
89 absl::Status SelectConcat(const ConcatAttributes& attr, in SelectConcat()
264 auto attr = absl::any_cast<ConcatAttributes>(node.operation.attributes); in GPUOperationFromNode()