Searched refs:seq_len_shape (Results 1 – 2 of 2) sorted by relevance
50 const int *seq_len_shape = inputs[5]->shape_; in GruInferShape() local51 if (seq_len_shape[0] > 1) { in GruInferShape()54 if (inputs[5]->shape_size_ != 1 && seq_len_shape[0] != in_shape[1]) { in GruInferShape()
82 auto seq_len_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1); in Init() local83 is_null_input_ = CHECK_NULL_INPUT(input_shape_) || CHECK_NULL_INPUT(seq_len_shape); in Init()99 seq_len_size_ = seq_len_shape.size(); in Init()