cavis/libnd4j/include/graph/scheme/config.fbs

57 lines
2.3 KiB
Plaintext

/*******************************************************************************
* Copyright (c) 2015-2018 Skymind, Inc.
*
* This program and the accompanying materials are made available under the
* terms of the Apache License, Version 2.0 which is available at
* https://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.
*
* SPDX-License-Identifier: Apache-2.0
******************************************************************************/
namespace nd4j.graph;
enum ProfilingMode:byte {
NONE, // no checks for Z values
NAN_PANIC, // reporting nodes with NaN values in Z
INF_PANIC, // reporting nodes with Inf values in Z
ANY_PANIC, // reporting nodes with NaN/Inf values in Z
}
enum ExecutionMode:byte {
SEQUENTIAL, // all operations will be executed sequentially
STRICT, // all operations will be following device id for execution mode selection
AUTO, // all operations that can be executed in parallel - will be executed in parallel
}
enum OutputMode:byte {
IMPLICIT, // only final nodes of graph will be returned
EXPLICIT, // only declared output fields
EXPLICIT_AND_IMPLICIT, // both options enabled
VARIABLE_SPACE, // whole content of VariableSpace will be returned back
OPTIMIZED, // same as IMPLICIT + using in-place ops execution where possible
}
enum Direction:byte {
FORWARD_ONLY, // inference pass
FORWARD_AND_BACKWARD, // not implemented yet
BACKWARD_ONLY, // not implemented yet
}
table FlatConfiguration {
id:long; // id of the graph this configuration applies to
executionMode:ExecutionMode; // current execution dmode
profilingMode:ProfilingMode; // current profiling mode
outputMode:OutputMode; // current output mode
timestats:bool; // are we gathering time info. false by default
footprintForward:long; // optional: amount of memory needed for FF pass
footprintBackward:long; // optional: amount of memory needed for BP pass
direction:Direction; // execution direction
}
root_type FlatConfiguration;