Searched refs:TransposeAttributes (Results 1 – 9 of 9) sorted by relevance
28 const TransposeAttributes& attr) { in GetTransposeCode()100 const TransposeAttributes& attr) { in CreateTranspose()
27 const TransposeAttributes& attr);
34 TransposeAttributes attr; in TransposeTest()
528 struct TransposeAttributes { struct535 BHWC CalculateOutputShape(const BHWC& input, const TransposeAttributes& attr); argument
803 BHWC CalculateOutputShape(const BHWC& input, const TransposeAttributes& attr) { in CalculateOutputShape()
2146 TransposeAttributes attr; in Parse()
89 void SelectTranspose(const TransposeAttributes& attr,
181 void SelectTranspose(const TransposeAttributes& attr, in SelectTranspose()
257 TransposeAttributes transpose_attr; in GPUOperationFromNode()501 absl::any_cast<TransposeAttributes>(node.operation.attributes); in GPUOperationFromNode()