Searched defs:kInputShape (Results 1 – 4 of 4) sorted by relevance
33 constexpr auto kInputShape = "[input_shape]"; variable
74 int64_t kInputShape = batch_rank; in ApplyAdaMaxInferShape() local
117 constexpr auto kInputShape = "input_shape"; variable
33 constexpr int64_t kInputShape = 2; variable