Searched defs:axis_size (Results 1 – 9 of 9) sorted by relevance
96 int axis_size = input_shape.Dims(axis); in StartForAxis() local145 const int axis_size = input_shape.Dims(axis); in StopForAxis() local
55 const int64 axis_size = input_shape.dim_size(axis); in Compile() local
63 const int64 axis_size = input_shape.dim_size(axis); in Compile() local
164 const int64 axis_size = ShapeUtil::GetDimension(input_shape, axis); in ArgMinMax() local
359 axis_size = lambda axis: -1 if axis.size is None else axis.size function
957 FT_UShort axis_size; in sfnt_init_face() local
78 TfLiteIntArray* axis_size = TfLiteIntArrayCreate(1); in ResizeTempAxis() local
1985 FT_Offset axis_size; in TT_Get_MM_Var() local
2947 const int axis_size = input_shape.Dims(axis); in Gather() local3878 const int axis_size = input1_shape.Dims(axis); in ArgMinMax() local