Searched refs:fw_output_shape (Results 1 – 1 of 1) sorted by relevance
1221 Shape* fw_output_shape = fw_output_array.mutable_shape(); in ProcessBidirectionalSequenceLstmOperator() local1223 fw_output_shape->ReplaceDims({timestamp, batch_size, 2 * output_size}); in ProcessBidirectionalSequenceLstmOperator()1225 fw_output_shape->ReplaceDims({timestamp, batch_size, output_size}); in ProcessBidirectionalSequenceLstmOperator()1273 Shape* fw_output_shape = fw_output_array.mutable_shape(); in ProcessBidirectionalSequenceRnnOperator() local1275 fw_output_shape->ReplaceDims({timestamp, batch_size, 2 * output_size}); in ProcessBidirectionalSequenceRnnOperator()1277 fw_output_shape->ReplaceDims({timestamp, batch_size, output_size}); in ProcessBidirectionalSequenceRnnOperator()