Searched refs:masked_input (Results 1 – 5 of 5) sorted by relevance
193 auto masked_input = Select( in TorchGather() local197 return Reduce(masked_input, Zero(builder, input_shape.element_type()), in TorchGather()
221 define i25 @masked_input(i25 %A) {222 ; CHECK-LABEL: @masked_input(
178 masked_input = keras.layers.Masking()(inputs)182 output = gru_layer(masked_input)194 cudnn_model = keras.models.Model(inputs, cudnn_layer(masked_input))
351 masked_input = keras.layers.Masking()(inputs)354 output = lstm_layer(masked_input)364 cudnn_model = keras.models.Model(inputs, cudnn_layer(masked_input))
1351 masked_input = mask(x)1352 y = layer(masked_input)