Searched defs:kDims (Results 1 – 3 of 3) sorted by relevance
76 constexpr int kDims = is_int8x4 ? 5 : 4; in Compute() local
108 const int kDims = input_shape.dimensions_count(); in ExportFloatArray() local2429 const int kDims = state_shape.dimensions_count(); in AddPlaceholderForRNNState() local