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