Turi Create  4.0
turi::neural_net::model_spec Member List

This is the complete list of members for turi::neural_net::model_spec, including all inherited members.

add_add_broadcastable(const std::string &name, const std::vector< std::string > &inputs)turi::neural_net::model_spec
add_addition(const std::string &name, const std::vector< std::string > &inputs)turi::neural_net::model_spec
add_batchnorm(const std::string &name, const std::string &input, size_t num_channels, float epsilon)turi::neural_net::model_spec
add_channel_concat(const std::string &name, const std::vector< std::string > &inputs)turi::neural_net::model_spec
add_channel_slice(const std::string &name, const std::string &input, int start_index, int end_index, size_t stride)turi::neural_net::model_spec
add_concat_nd(const std::string &name, const std::vector< std::string > &inputs, size_t axis)turi::neural_net::model_spec
add_constant(const std::string &name, const std::array< size_t, 3 > &shape_c_h_w, weight_initializer weight_initializer_fn)turi::neural_net::model_spec
add_constant_nd(const std::string &name, const std::vector< size_t > &shape, const weight_initializer &data)turi::neural_net::model_spec
add_convolution(const std::string &name, const std::string &input, size_t num_output_channels, size_t num_kernel_channels, size_t kernel_height, size_t kernel_width, size_t stride_h, size_t stride_w, padding_type padding, weight_initializer weight_initializer_fn, weight_initializer bias_initializer_fn=nullptr)turi::neural_net::model_spec
add_exp(const std::string &name, const std::string &input)turi::neural_net::model_spec
add_expand_dims(const std::string &name, const std::string &input, const std::vector< size_t > &axes, const std::vector< size_t > &inputVector, const std::vector< size_t > &outputVector)turi::neural_net::model_spec
add_flatten(const std::string &name, const std::string &input)turi::neural_net::model_spec
add_gather(const std::string &name, const std::vector< std::string > &inputs)turi::neural_net::model_spec
add_get_shape(const std::string &name, const std::string &input)turi::neural_net::model_spec
add_inner_product(const std::string &name, const std::string &input, size_t num_output_channels, size_t num_input_channels, weight_initializer weight_initializer_fn, weight_initializer bias_initializer_fn=nullptr)turi::neural_net::model_spec
add_instancenorm(const std::string &name, const std::string &input, size_t num_channels, float epsilon)turi::neural_net::model_spec
add_leakyrelu(const std::string &name, const std::string &input, float alpha)turi::neural_net::model_spec
add_lstm(const std::string &name, const std::string &input, const std::string &hidden_input, const std::string &cell_input, const std::string &hidden_output, const std::string &cell_output, size_t input_vector_size, size_t output_vector_size, float cell_clip_threshold, const lstm_weight_initializers &initializers)turi::neural_net::model_spec
add_multiplication(const std::string &name, const std::vector< std::string > &inputs)turi::neural_net::model_spec
add_nms_layer(const std::string &name, const std::vector< std::string > &inputs, const std::vector< std::string > &outputs, float iou_threshold, float confidence_threshold, size_t max_boxes, bool per_class_supression)turi::neural_net::model_spec
add_padding(const std::string &name, const std::string &input, size_t padding_top, size_t padding_bottom, size_t padding_left, size_t padding_right, padding_policy policy=padding_policy::REFLECTIVE)turi::neural_net::model_spec
add_permute(const std::string &name, const std::string &input, const std::array< size_t, 4 > &axis_permutation)turi::neural_net::model_spec
add_pooling(const std::string &name, const std::string &input, size_t kernel_height, size_t kernel_width, size_t stride_h, size_t stride_w, padding_type padding, bool use_poolexcludepadding=false, pooling_type pooling=pooling_type::MAX)turi::neural_net::model_spec
add_preprocessing(const std::string &feature_name, const float image_scale)turi::neural_net::model_spec
add_relu(const std::string &name, const std::string &input)turi::neural_net::model_spec
add_reshape(const std::string &name, const std::string &input, const std::array< size_t, 4 > &seq_c_h_w)turi::neural_net::model_spec
add_reshape_dynamic(const std::string &name, const std::vector< std::string > &inputs)turi::neural_net::model_spec
add_reshape_static(const std::string &name, const std::string &input, const std::vector< size_t > &targetShape)turi::neural_net::model_spec
add_scale(const std::string &name, const std::string &input, const std::vector< size_t > &shape_c_h_w, weight_initializer scale_initializer_fn)turi::neural_net::model_spec
add_sigmoid(const std::string &name, const std::string &input)turi::neural_net::model_spec
add_slice_dynamic(const std::string &name, const std::vector< std::string > &inputs)turi::neural_net::model_spec
add_softmax(const std::string &name, const std::string &input)turi::neural_net::model_spec
add_split_nd(const std::string &name, const std::string &input, size_t axis, size_t num_splits, const std::vector< size_t > &split_sizes)turi::neural_net::model_spec
add_squeeze(const std::string &name, const std::string &input, const std::vector< size_t > &axes, const std::vector< size_t > &inputVector, const std::vector< size_t > &outputVector)turi::neural_net::model_spec
add_transpose(const std::string &name, const std::string &input, std::vector< size_t > axes)turi::neural_net::model_spec
add_upsampling(const std::string &name, const std::string &input, size_t scaling_x, size_t scaling_y)turi::neural_net::model_spec
export_params_view() constturi::neural_net::model_spec
get_coreml_spec() constturi::neural_net::model_specinline
has_layer_output(const std::string &layer_name) constturi::neural_net::model_spec
model_spec()turi::neural_net::model_spec
model_spec(const CoreML::Specification::NeuralNetwork &nn_model)turi::neural_net::model_spec
model_spec(const std::string &mlmodel_path)turi::neural_net::model_spec
move_coreml_spec() &&turi::neural_net::model_spec
padding_policy enum nameturi::neural_net::model_spec
padding_type enum nameturi::neural_net::model_spec
pooling_type enum nameturi::neural_net::model_spec
update_params(const float_array_map &weights, bool use_quantization=false)turi::neural_net::model_spec