Aapo Kyrola dcefc74a0c Shape and Type Inference Part1 ...

Summary:
This is a bit large diff, sorry about it. It includes basic shape and type inference functionality, based on YQ's Schema scaffolding. I added some helper functions to make it easier to write simple translations.

Bigger refactoring was needed for ConvPoolBase so that we could use the shape inference already there in the schema.

I annotated enough operators to be able to infer forward-pass of shapes for basic convnet, and added test for that. I intend to bootcamp some annotations and annotate enough to handle Resnets fully. Need to think about gradients, if they could be annotated in an easier way.

Only shapes are now exposed to Python, types will follow later. Also the inference is not called yet anywhere but unit test.

Also I am not sure if everything is in the best location in the code, but shouldn't be hard to move stuff around.

Reviewed By: dzhulgakov

Differential Revision: D4436818

fbshipit-source-id: eebee5937ccc9ac09c245465302388a1fae6933c

2017-02-02 22:29:22 -08:00

..

activation_ops_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

atomic_ops_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

checkpoint_test.py

snapshot -> checkpoint

2016-12-15 12:01:30 -08:00

conv_test.py

Add error checking for too-small input in ConvPoolOpBase

2017-01-25 17:44:22 -08:00

conv_transpose_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

copy_ops_test.py

gradients for CopyGPUToCPU and CopyCPUToGPU + unit test + schema

2016-12-15 12:01:29 -08:00

cosine_embedding_criterion_op_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

counter_ops_test.py

AtomicCounter to return previous value on Reset.

2017-02-02 14:59:30 -08:00

cross_entropy_ops_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

dataset_ops_test.py

Last N window collector

2016-12-15 12:01:30 -08:00

duplicate_operands_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

elementwise_op_broadcast_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

elementwise_ops_test.py

add CUDA gradient for Div

2017-01-09 21:59:23 -08:00

emptysample_ops_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

extend_tensor_op_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

fc_operator_test.py

Test for FC operator + fix for docs

2017-01-27 10:44:24 -08:00

gather_ops_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

gather_ranges_op_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

group_conv_test.py

Make all convolution operators allow optional bias term

2016-12-21 15:14:24 -08:00

hsm_test.py

Generate huffman tree

2017-01-19 16:14:23 -08:00

index_ops_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

instance_norm_test.py

instance norm

2017-01-20 12:29:28 -08:00

margin_ranking_criterion_op_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

matmul_op_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

mkl_conv_op_test.py

MKL convolution operator

2017-01-23 09:59:30 -08:00

mkl_packed_fc_op_test.py

MKL convolution operator

2017-01-23 09:59:30 -08:00

mkl_speed_test.py

MKL convolution operator

2017-01-23 09:59:30 -08:00

momentum_sgd_test.py

MomemtumSGDUpdate -- version of MomentumSGD with update.

2016-12-15 12:01:29 -08:00

mpi_test.py

Setup MPI before test start

2016-12-19 15:59:32 -08:00

one_hot_ops_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

pack_ops_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

partition_ops_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

pooling_test.py

Unit test for big batch size avg pooling

2017-01-18 19:29:20 -08:00

python_op_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

rank_loss_operator_test.py

Normalize rank loss gradient to avoid convergence issues when the number of pairs is really large

2016-12-21 17:29:24 -08:00

record_queue_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

recurrent_network_test.py

remove recurrent_inputs in a favor of recurrent_input_ids

2017-01-31 13:14:33 -08:00

relu_op_test.py

MKLDevice and MKLOperator

2016-12-15 19:59:24 -08:00

reshape_ops_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

segment_ops_test.py

Incremental MeanReducer segment Ops

2016-11-29 15:18:38 -08:00

sequence_ops_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

shape_inference_test.py

Shape and Type Inference Part1

2017-02-02 22:29:22 -08:00

softmax_ops_test.py

softmax-with-loss, handle gracefully cases when total weight is 0

2017-01-20 15:29:21 -08:00

sparse_gradient_checker_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

sparse_ops_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

spatial_bn_op_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

square_root_divide_op_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

string_ops_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00

text_file_reader_test.py

Add unittest.main() lines to test scripts under python/operator_test

2016-11-29 15:18:37 -08:00