blob: 568edbaa1e8c19d69bdde91fdb484206a52752d8 [file] [log] [blame]
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements. See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to you under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
source("tests/onnx_systemds/dml_output/simple_dropout_layer.dml") as simple_dropout
A = matrix("3 8 9 10", rows=2, cols=2)
[O, mask] = simple_dropout::gen_simple_dropout_graph(A)
out = append(toString(O), toString(mask))
write(out, "tests/onnx_systemds/output_test/simple_dropout_layer.out")