Searched defs:output_cell_shape (Results 1 – 1 of 1) sorted by relevance
50 …orShape recurrent_weights_shape, TensorShape cell_bias_shape, TensorShape output_cell_shape, Tenso… in setup()80 …const TensorShape &output_cell_shape, const TensorShape &output_shape, const TensorShape &scratch_… in compute_target()325 …const TensorShape &output_cell_shape, const TensorShape &output_shape, const TensorShape &scratch_… in compute_reference()