Searched defs:axis_size (Results 1 – 15 of 15) sorted by relevance
80 const int axis_size = input_shape.Dims(axis); in StartForAxis() local129 const int axis_size = input_shape.Dims(axis); in StopForAxis() local
37 const int axis_size = input_shape.Dims(axis); in Gather() local
35 const int axis_size = input1_shape.Dims(axis); in ArgMinMax() local
57 const int64 axis_size = input_shape.dim_size(axis); in Compile() local
73 xla::XlaOp axis_size = xla::ConstantR0WithType( in Compile() local
42 constexpr int axis_size = 1; in TestSplitTwoOutputsFloat() local103 constexpr int axis_size = 1; in TestSplitFourOutputsFloat() local170 constexpr int axis_size = 1; in TestSplitTwoOutputsQuantized() local226 constexpr int axis_size = 1; in TestSplitTwoOutputsQuantized32() local
53 constexpr int axis_size = 1; in TestSplitVFloat() local
143 size_t axis_size = rmins.size(); in fakeQuantAttrsToType() local
585 FT_UShort axis_size; in sfnt_init_face() local
144 const int* curr_pos, const int* axis, const int axis_size) { in GetOffset()
84 TfLiteIntArray* axis_size = TfLiteIntArrayCreate(1); in ResizeTempAxis() local
381 size_t axis_size = 1; in GetUniformQuantizedTypeForBias() local
316 FT_Offset axis_size; in T1_Get_MM_Var() local
1475 const int axis_size = axis.size(); in IsAlongAxis() local1739 int axis_size = axis.size(); in IsAlongAxis() local
2060 FT_Offset axis_size; in TT_Get_MM_Var() local