Module ImportFrom alias ImportFrom alias Import alias Import alias Import alias Import alias identifier Import alias identifier Import alias identifier If UnaryOpNot AttributeLoad NameLoad attr body Raise Call NameLoad Str If CompareEq AttributeLoad AttributeLoad NameLoad attr attr Str body Assign NameStore Call AttributeLoad Call AttributeLoad AttributeLoad AttributeLoad NameLoad attr attr attr Str attr Str Assign NameStore Call AttributeLoad AttributeLoad AttributeLoad NameLoad attr attr attr Str orelse Assign NameStore Call AttributeLoad Call AttributeLoad AttributeLoad AttributeLoad NameLoad attr attr attr attr Str Assign NameStore Call AttributeLoad AttributeLoad AttributeLoad NameLoad attr attr attr FunctionDef arguments args defaults body Expr Str Assign NameStore Num Assign NameStore Num Assign NameStore Num If UnaryOpNot Call NameLoad NameLoad AttributeLoad AttributeLoad NameLoad attr attr body Assign NameStore Num Assign NameStore Num Assign NameStore Call AttributeLoad NameLoad attr Str Assign NameStore Call AttributeLoad NameLoad attr Str Assign NameStore Call AttributeLoad NameLoad attr Str Expr Call AttributeLoad NameLoad attr Assign NameStore Call AttributeLoad NameLoad attr Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad NameLoad keyword AttributeLoad NameLoad attr Assign NameStore Call AttributeLoad NameLoad attr TupleLoad NameLoad keyword Str Assign NameStore Call AttributeLoad NameLoad attr TupleLoad NameLoad keyword Str Assign NameStore Call AttributeLoad NameLoad attr Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad NameLoad keyword Str Assign NameStore Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr NameLoad NameLoad keyword Str Delete NameDel Assign NameStore Call AttributeLoad AttributeLoad NameLoad attr attr BinOpAdd NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr NameLoad keyword Num Assign NameStore UnaryOpUSub Call AttributeLoad NameLoad attr SubscriptLoad Call AttributeLoad NameLoad attr NameLoad Index TupleLoad Call AttributeLoad NameLoad attr SubscriptLoad AttributeLoad NameLoad attr Index Num NameLoad Assign NameStore Call AttributeLoad NameLoad attr NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr keyword ListLoad NameLoad NameLoad NameLoad keyword ListLoad NameLoad NameLoad NameLoad keyword NameLoad Assign NameStore Call AttributeLoad NameLoad attr keyword ListLoad NameLoad NameLoad NameLoad keyword ListLoad NameLoad NameLoad NameLoad keyword NameLoad Assert Call NameLoad ListComp Call NameLoad AttributeLoad NameLoad attr AttributeLoad AttributeLoad NameLoad attr attr comprehension NameStore Call AttributeLoad AttributeLoad AttributeLoad NameLoad attr attr attr Assert Call NameLoad ListComp Call NameLoad AttributeLoad NameLoad attr AttributeLoad AttributeLoad NameLoad attr attr comprehension NameStore Call AttributeLoad AttributeLoad AttributeLoad NameLoad attr attr attr Assign NameStore Call NameLoad NameLoad NameLoad NameLoad Assign NameStore Call NameLoad NameLoad NameLoad NameLoad Assert CompareEqEq Call NameLoad NameLoad Call NameLoad NameLoad Num Assert Call AttributeLoad NameLoad attr SubscriptLoad NameLoad Index Num SubscriptLoad NameLoad Index Num Assert Call AttributeLoad NameLoad attr SubscriptLoad NameLoad Index Num SubscriptLoad NameLoad Index Num keyword Num Call AttributeLoad Call AttributeLoad NameLoad attr BinOpSub NameLoad NameLoad attr Assert Call AttributeLoad NameLoad attr SubscriptLoad NameLoad Index Num SubscriptLoad NameLoad Index Num ListComp TupleLoad NameLoad SubscriptLoad SubscriptLoad NameLoad Index Num Index NameLoad SubscriptLoad SubscriptLoad NameLoad Index Num Index NameLoad NameLoad comprehension TupleStore NameStore NameStore Call NameLoad BinOpSub SubscriptLoad NameLoad Index Num SubscriptLoad NameLoad Index Num CompareNotEq NameLoad Num decorator_list FunctionDef arguments args defaults body Expr Str Assign NameStore Num Assign NameStore Num If UnaryOpNot Call NameLoad NameLoad AttributeLoad AttributeLoad NameLoad attr attr body Assign NameStore Num Expr Call AttributeLoad NameLoad attr Assign NameStore Call AttributeLoad Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad NameLoad attr Str Assign NameStore Call AttributeLoad NameLoad attr Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad keyword Str Assign NameStore Call AttributeLoad AttributeLoad NameLoad attr attr keyword Num keyword Num keyword NameLoad Assign NameStore Call AttributeLoad NameLoad attr AugAssignDiv NameStore Call AttributeLoad Call AttributeLoad NameLoad attr keyword Num attr SubscriptLoad AttributeLoad NameLoad attr Index Num Num Assign NameStore Call AttributeLoad AttributeLoad AttributeLoad NameLoad attr attr attr NameLoad NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr ListLoad NameLoad NameLoad keyword NameLoad Assign NameStore Call AttributeLoad NameLoad attr ListLoad NameLoad NameLoad keyword NameLoad Assert Call NameLoad ListComp Call NameLoad AttributeLoad NameLoad attr AttributeLoad AttributeLoad NameLoad attr attr comprehension NameStore Call AttributeLoad AttributeLoad AttributeLoad NameLoad attr attr attr Assert Call NameLoad ListComp Call NameLoad AttributeLoad NameLoad attr AttributeLoad AttributeLoad NameLoad attr attr comprehension NameStore Call AttributeLoad AttributeLoad AttributeLoad NameLoad attr attr attr Assign NameStore Call NameLoad NameLoad Assign NameStore Call NameLoad NameLoad Assign NameStore Num Assign NameStore Num If UnaryOpNot Call AttributeLoad NameLoad attr NameLoad NameLoad keyword NameLoad keyword NameLoad body Assign TupleStore NameStore NameStore Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr BinOpDiv NameLoad NameLoad BinOpDiv NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr Expr Call NameLoad Str NameLoad BinOpDiv NameLoad NameLoad keyword Str Expr Call NameLoad BinOpMod NameLoad NameLoad BinOpDiv NameLoad NameLoad BinOpBitAnd NameLoad NameLoad Expr Call NameLoad Str Expr Call NameLoad Str SubscriptLoad Call AttributeLoad NameLoad attr Index NameLoad Expr Call NameLoad Str SubscriptLoad Call AttributeLoad NameLoad attr Index NameLoad Expr Call NameLoad Str SubscriptLoad Call AttributeLoad NameLoad attr Index NameLoad Expr Call NameLoad Str SubscriptLoad Call AttributeLoad NameLoad attr Index NameLoad Expr Call NameLoad Str SubscriptLoad Call AttributeLoad NameLoad attr Index NameLoad Expr Call NameLoad Str SubscriptLoad Call AttributeLoad NameLoad attr Index NameLoad Expr Call NameLoad Str SubscriptLoad NameLoad Index BinOpDiv NameLoad NameLoad Expr Call NameLoad Str SubscriptLoad NameLoad Index BinOpDiv NameLoad NameLoad Assert NameLoad BinOpMod Str TupleLoad NameLoad NameLoad decorator_list FunctionDef arguments args defaults body Expr Str Assign NameStore Call AttributeLoad NameLoad attr Str Assign NameStore Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad SubscriptLoad Call AttributeLoad NameLoad attr BinOpMult SubscriptLoad AttributeLoad NameLoad attr Index Num Num keyword Str Slice Num Assign NameStore Call AttributeLoad NameLoad attr ListLoad NameLoad NameLoad keyword NameLoad Assign NameStore Call AttributeLoad NameLoad attr ListLoad NameLoad NameLoad keyword NameLoad Assert CompareEq AttributeLoad SubscriptLoad Call AttributeLoad AttributeLoad AttributeLoad NameLoad attr attr attr Index Num attr AttributeLoad AttributeLoad NameLoad attr attr Assert Call NameLoad AttributeLoad SubscriptLoad Call AttributeLoad AttributeLoad AttributeLoad NameLoad attr attr attr Index Num attr AttributeLoad AttributeLoad NameLoad attr attr FunctionDef arguments args NameParam NameParam defaults body Assign NameStore Call AttributeLoad Call AttributeLoad NameLoad attr BinOpMult NameLoad NameLoad keyword Str attr NameLoad NameLoad Assign NameStore Call NameLoad NameLoad Assign NameStore Call NameLoad NameLoad Assert Call AttributeLoad NameLoad attr NameLoad NameLoad Call AttributeLoad NameLoad attr BinOpSub NameLoad NameLoad decorator_list Expr Call NameLoad Num Num Expr Call NameLoad BinOpLShift Num Num Num Expr Call NameLoad Num Num Expr Call NameLoad Num Num Expr Call NameLoad Num Num Expr Call NameLoad Num Num Expr Call NameLoad Num Num Expr Call NameLoad Num Num Expr Call NameLoad Num Num Expr Call NameLoad Num Num Expr Call NameLoad Num Num Expr Call NameLoad Num BinOpMult Num Num Expr Call NameLoad Num BinOpMult Num Num decorator_list ClassDef bases AttributeLoad NameLoad attr body Assign NameStore AttributeLoad AttributeLoad NameLoad attr attr Assign NameStore Call AttributeLoad NameLoad attr Str Assign NameStore NameLoad Assign NameStore NameLoad Assign NameStore Num FunctionDef arguments args NameParam NameParam NameParam NameParam NameParam NameParam NameParam defaults body Expr Str Assign NameStore Call NameLoad NameLoad Assign NameStore Call NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr ListLoad NameLoad NameLoad keyword NameLoad Assign NameStore Call AttributeLoad NameLoad attr ListLoad NameLoad NameLoad keyword AttributeLoad NameLoad attr Expr Call NameLoad NameLoad NameLoad Expr Call NameLoad Num Num NameLoad NameLoad Expr Call NameLoad Num Num NameLoad NameLoad Expr Call NameLoad Num Num NameLoad NameLoad Expr Call NameLoad Num Num NameLoad NameLoad Expr Call NameLoad Num Num NameLoad NameLoad Expr Call NameLoad Num Num NameLoad NameLoad Expr Call NameLoad Num Num NameLoad NameLoad Expr Call NameLoad Num Num NameLoad NameLoad Expr Call NameLoad Num Num NameLoad NameLoad Expr Call NameLoad Num Num NameLoad NameLoad Expr Call NameLoad Num Num NameLoad NameLoad Expr Call NameLoad Num Num NameLoad NameLoad Expr Call NameLoad Num Num NameLoad NameLoad Expr Call NameLoad Num Num NameLoad NameLoad Expr Call NameLoad Num BinOpMult Num Num NameLoad NameLoad Expr Call NameLoad Num BinOpMult Num Num NameLoad NameLoad Expr Call NameLoad BinOpSub BinOpLShift Num Num Num Num NameLoad NameLoad Expr Call NameLoad Num BinOpLShift Num Num NameLoad NameLoad Return TupleLoad NameLoad NameLoad decorator_list FunctionDef arguments args NameParam NameParam NameParam NameParam NameParam defaults body Assign NameStore Call AttributeLoad Call AttributeLoad NameLoad attr BinOpMult NameLoad NameLoad keyword Str attr NameLoad NameLoad Assign NameStore Call NameLoad NameLoad Assign NameStore Call NameLoad NameLoad Expr Call AttributeLoad NameLoad attr NameLoad NameLoad decorator_list FunctionDef arguments args NameParam NameParam NameParam NameParam NameParam defaults body Assert Call NameLoad AttributeLoad SubscriptLoad Call AttributeLoad AttributeLoad AttributeLoad NameLoad attr attr attr Index Num attr NameLoad Assert Call NameLoad AttributeLoad SubscriptLoad Call AttributeLoad AttributeLoad AttributeLoad NameLoad attr attr attr Index AttributeLoad NameLoad attr attr NameLoad decorator_list FunctionDef arguments args NameParam defaults body Assign NameStore Call AttributeLoad NameLoad attr Str Assign NameStore AttributeLoad AttributeLoad NameLoad attr attr FunctionDef arguments args NameParam NameParam defaults body Expr Call AttributeLoad NameLoad attr NameLoad NameLoad Call NameLoad NameLoad AttributeLoad NameLoad attr decorator_list Assign TupleStore NameStore NameStore Call AttributeLoad NameLoad attr NameLoad NameLoad NameLoad NameLoad AttributeLoad NameLoad attr NameLoad If AttributeLoad NameLoad attr body Expr Call AttributeLoad NameLoad attr BinOpLShift Num Num Num NameLoad NameLoad If AttributeLoad NameLoad attr body Expr Call AttributeLoad NameLoad attr Num Num NameLoad NameLoad decorator_list decorator_list
N/A __future__ print_function nose.plugins.skip SkipTest N/A numpy N/A unittest N/A theano N/A theano.tensor T N/A theano.tests.unittest_tools utt N/A theano.sandbox.cuda cuda N/A N/A N/A cuda cuda_available N/A N/A N/A SkipTest Optional package cuda disabled N/A N/A N/A N/A theano config mode FAST_COMPILE N/A N/A mode_with_gpu N/A N/A N/A N/A N/A N/A theano compile mode get_mode FAST_RUN including gpu N/A mode_without_gpu N/A N/A N/A N/A theano compile mode get_mode FAST_RUN N/A N/A mode_with_gpu N/A N/A N/A N/A N/A N/A theano compile mode get_default_mode including gpu N/A mode_without_gpu N/A N/A N/A N/A theano compile mode get_default_mode test_GpuCrossentropySoftmaxArgmax1HotWithBias N/A N/A N/A N/A N/A 
    This is basic test for GpuCrossentropySoftmaxArgmax1HotWithBias

    We check that we loop when there are too many threads

     N/A n_in 1000 N/A batch_size 4097 N/A n_out 1250 N/A N/A N/A isinstance mode_with_gpu N/A N/A theano compile DebugMode N/A N/A n_in 4098 N/A n_out 4099 N/A y N/A N/A T lvector y N/A b N/A N/A T fvector b N/A dot_result N/A N/A T fmatrix dot_result N/A N/A N/A utt seed_rng N/A xx N/A N/A numpy asarray N/A N/A N/A numpy random rand batch_size n_in dtype N/A numpy float32 N/A yy N/A N/A numpy ones N/A batch_size dtype int32 N/A b_values N/A N/A numpy zeros N/A n_out dtype float32 N/A W_values N/A N/A numpy asarray N/A N/A N/A numpy random rand n_in n_out dtype float32 N/A dot_value N/A N/A numpy asarray N/A N/A numpy dot xx W_values dtype float32 N/A W_values N/A p_y_given_x N/A N/A N/A T nnet softmax N/A dot_result b N/A y_pred N/A N/A T argmax p_y_given_x axis -1 N/A loss N/A N/A N/A T mean N/A N/A N/A T log p_y_given_x N/A N/A N/A N/A T arange N/A N/A y shape N/A 0 y N/A dW N/A N/A T grad loss dot_result N/A classify N/A N/A theano function inputs N/A y b dot_result outputs N/A loss y_pred dW mode mode_without_gpu N/A classify_gpu N/A N/A theano function inputs N/A y b dot_result outputs N/A loss y_pred dW mode mode_with_gpu N/A N/A any N/A N/A isinstance N/A node op N/A N/A T nnet CrossentropySoftmaxArgmax1HotWithBias N/A node N/A N/A N/A N/A classify maker fgraph toposort N/A N/A any N/A N/A isinstance N/A node op N/A N/A cuda nnet GpuCrossentropySoftmaxArgmax1HotWithBias N/A node N/A N/A N/A N/A classify_gpu maker fgraph toposort N/A out N/A classify yy b_values dot_value N/A gout N/A classify_gpu yy b_values dot_value N/A N/A N/A len out N/A len gout 3 N/A N/A N/A numpy allclose N/A out N/A 0 N/A gout N/A 0 N/A N/A N/A numpy allclose N/A out N/A 2 N/A gout N/A 2 atol 3e-06 N/A N/A N/A N/A numpy absolute N/A gout out max N/A N/A N/A numpy allclose N/A out N/A 1 N/A gout N/A 1 N/A N/A id N/A N/A out N/A 1 N/A id N/A N/A gout N/A 1 N/A id val N/A N/A id val N/A enumerate N/A N/A out N/A 1 N/A gout N/A 1 N/A val 0 N/A test_GpuCrossentropySoftmax1HotWithBiasDx N/A N/A N/A N/A N/A 
    This is basic test for GpuCrossentropySoftmax1HotWithBiasDx

    We check that we loop when there are too many threads

     N/A batch_size 4097 N/A n_out 1250 N/A N/A N/A isinstance mode_with_gpu N/A N/A theano compile DebugMode N/A N/A n_out 4099 N/A N/A N/A utt seed_rng N/A softmax_output_value N/A N/A N/A N/A N/A numpy random rand batch_size n_out astype float32 N/A dnll_value N/A N/A numpy asarray N/A N/A N/A numpy random rand batch_size dtype float32 N/A y_idx_value N/A N/A N/A numpy random randint low 0 high 5 size batch_size N/A softmax_output N/A N/A T fmatrix N/A softmax_output N/A N/A N/A N/A softmax_output sum axis 1 reshape N/A N/A softmax_output shape N/A 1 1 N/A op N/A N/A N/A N/A theano tensor nnet crossentropy_softmax_1hot_with_bias_dx dnll_value softmax_output y_idx_value N/A cpu_f N/A N/A theano function N/A softmax_output op mode mode_without_gpu N/A gpu_f N/A N/A theano function N/A softmax_output op mode mode_with_gpu N/A N/A any N/A N/A isinstance N/A node op N/A N/A T nnet CrossentropySoftmax1HotWithBiasDx N/A node N/A N/A N/A N/A cpu_f maker fgraph toposort N/A N/A any N/A N/A isinstance N/A node op N/A N/A cuda nnet GpuCrossentropySoftmax1HotWithBiasDx N/A node N/A N/A N/A N/A gpu_f maker fgraph toposort N/A cpu_out N/A cpu_f softmax_output_value N/A gpu_out N/A gpu_f softmax_output_value N/A rtol 1e-05 N/A atol 1e-06 N/A N/A N/A N/A numpy allclose cpu_out gpu_out rtol rtol atol atol N/A N/A N/A abs_err rel_err N/A N/A N/A T numeric_grad abs_rel_err cpu_out gpu_out N/A scaled_err N/A N/A numpy minimum N/A abs_err atol N/A rel_err rtol N/A max_i N/A N/A scaled_err argmax N/A N/A print max err index: max_i N/A max_i batch_size end N/A N/A print N/A max_i batch_size N/A max_i n_out N/A max_i n_out N/A N/A print At that index: N/A N/A print err: N/A N/A N/A scaled_err flatten N/A max_i N/A N/A print absolute error: N/A N/A N/A abs_err flatten N/A max_i N/A N/A print relative error: N/A N/A N/A rel_err flatten N/A max_i N/A N/A print cpu_out: N/A N/A N/A cpu_out flatten N/A max_i N/A N/A print gpu_out: N/A N/A N/A gpu_out flatten N/A max_i N/A N/A print softmax_output_value: N/A N/A N/A softmax_output_value flatten N/A max_i N/A N/A print dnll_value: N/A dnll_value N/A N/A max_i n_out N/A N/A print y_idx_value: N/A y_idx_value N/A N/A max_i n_out N/A False N/A numpy.allclose(cpu_out, gpu_out, rtol=%s, atol=%s) N/A rtol atol N/A test_softmax_with_bias N/A N/A N/A N/A N/A 
    This is basic test for GpuSoftmaxWithBias

    We check that we loop when their is too much block

    TODO: check that we loop when there are too many threads.(THIS IS
    NOT IMPLEMENTED)
     N/A x N/A N/A T fmatrix x N/A z N/A N/A N/A T nnet softmax_with_bias x N/A N/A N/A T arange N/A N/A N/A x shape N/A 1 2 dtype float32 N/A 2 N/A f N/A N/A theano function N/A x z mode mode_without_gpu N/A f_gpu N/A N/A theano function N/A x z mode mode_with_gpu N/A N/A N/A N/A N/A N/A N/A N/A f maker fgraph toposort N/A -1 op N/A N/A T nnet softmax_with_bias N/A N/A isinstance N/A N/A N/A N/A N/A N/A f_gpu maker fgraph toposort N/A -2 op N/A N/A cuda nnet GpuSoftmaxWithBias cmp N/A N/A n m N/A N/A N/A data N/A N/A N/A N/A numpy arange N/A n m dtype float32 reshape n m N/A out N/A f data N/A gout N/A f_gpu data N/A N/A N/A numpy allclose out gout N/A N/A numpy absolute N/A out gout N/A N/A N/A cmp 2 5 N/A N/A cmp N/A 2 15 5 N/A N/A cmp 4074 400 N/A N/A cmp 0 10 N/A N/A cmp 784 784 N/A N/A cmp 4 1000 N/A N/A cmp 4 1024 N/A N/A cmp 4 2000 N/A N/A cmp 4 2024 N/A N/A cmp 4 4074 N/A N/A cmp 2 10000 N/A N/A cmp 128 N/A 16 1024 N/A N/A cmp 128 N/A 64 1024 N/A test_SoftMax N/A N/A unittest TestCase N/A N/A gpu_op N/A N/A cuda nnet GpuSoftmax N/A mode N/A N/A mode_with_gpu excluding cudnn N/A do_big True N/A do_0 True N/A topo_idx -2 _test_softmax N/A N/A self x x_gpu f_z f_gpu_z cmp check_types N/A N/A N/A 
        This is basic test for GpuSoftmax and GpuDnnSoftmax

        We check that we loop when there is too much block
        We use slower code when there isn't enough shared memory
         N/A f_z_out N/A f_z x N/A f_gpu_z_out N/A f_gpu_z x_gpu N/A f N/A N/A theano function N/A x f_z_out mode mode_without_gpu N/A f_gpu N/A N/A theano function N/A x_gpu f_gpu_z_out mode N/A self mode N/A N/A check_types f f_gpu N/A N/A cmp 1 5 f f_gpu N/A N/A cmp 2 5 f f_gpu N/A N/A cmp 10 5 f f_gpu N/A N/A cmp 100 5 f f_gpu N/A N/A cmp 1000 5 f f_gpu N/A N/A cmp 10000 5 f f_gpu N/A N/A cmp 4074 400 f f_gpu N/A N/A cmp 784 784 f f_gpu N/A N/A cmp 4 1000 f f_gpu N/A N/A cmp 4 1024 f f_gpu N/A N/A cmp 4 2000 f f_gpu N/A N/A cmp 4 2024 f f_gpu N/A N/A cmp 4 4074 f f_gpu N/A N/A cmp 2 10000 f f_gpu N/A N/A cmp 128 N/A 16 1024 f f_gpu N/A N/A cmp 128 N/A 64 1024 f f_gpu N/A N/A cmp N/A N/A 2 15 1 5 f f_gpu N/A N/A cmp 5 N/A 2 15 f f_gpu N/A N/A f f_gpu N/A _cmp N/A N/A self n m f f_gpu N/A N/A N/A data N/A N/A N/A N/A numpy arange N/A n m dtype float32 reshape n m N/A out N/A f data N/A gout N/A f_gpu data N/A N/A N/A utt assert_allclose out gout N/A _check_types N/A N/A self graph graph_gpu f_type f_gpu_type N/A N/A N/A N/A isinstance N/A N/A N/A N/A N/A N/A graph maker fgraph toposort N/A -1 op f_type N/A N/A isinstance N/A N/A N/A N/A N/A N/A graph_gpu maker fgraph toposort N/A N/A self topo_idx op f_gpu_type N/A test_softmax N/A N/A self N/A N/A N/A x N/A N/A T fmatrix x N/A z N/A N/A T nnet softmax_op check_types N/A N/A graph graph_gpu N/A N/A N/A N/A N/A self _check_types graph graph_gpu N/A type z N/A self gpu_op N/A N/A N/A f f_gpu N/A N/A self _test_softmax x x z z N/A self _cmp check_types N/A N/A self do_big N/A N/A N/A N/A self _cmp N/A 2 15 5 f f_gpu N/A N/A self do_0 N/A N/A N/A N/A self _cmp 0 10 f f_gpu N/A N/A