Searched refs:bw_output_shape (Results 1 – 1 of 1) sorted by relevance
1213 Shape* bw_output_shape = bw_output_array.mutable_shape(); in ProcessBidirectionalSequenceLstmOperator() local1214 bw_output_shape->ReplaceDims({timestamp, batch_size, output_size}); in ProcessBidirectionalSequenceLstmOperator()1265 Shape* bw_output_shape = bw_output_array.mutable_shape(); in ProcessBidirectionalSequenceRnnOperator() local1266 bw_output_shape->ReplaceDims({timestamp, batch_size, output_size}); in ProcessBidirectionalSequenceRnnOperator()