Searched refs:GetHLOAxisFromTFAxis (Results 1 – 2 of 2) sorted by relevance
44 def GetHLOAxisFromTFAxis : NativeCodeCall<45 "GetHLOAxisFromTFAxis("51 "GetHLOAxisFromTFAxis("329 (GetHLOAxisFromTFAxis $axis, $params),330 (GetHLOAxisFromTFAxis $batch_dims, $indices))>;
195 static IntegerAttr GetHLOAxisFromTFAxis(Attribute attr, int64_t rank, in GetHLOAxisFromTFAxis() function