/usr/local/lib64/python3.6/site-packages/caffe2/python/__pycache__
NameSizeModeActions
allcompare_test.cpython-36.pyc27100644editdlrm
attention.cpython-36.pyc49380644editdlrm
benchmark_generator.cpython-36.pyc39100644editdlrm
binarysize.cpython-36.pyc48530644editdlrm
brew.cpython-36.pyc38490644editdlrm
brew_test.cpython-36.pyc104520644editdlrm
build.cpython-36.pyc3400644editdlrm
cached_reader.cpython-36.pyc29540644editdlrm
caffe_translator.cpython-36.pyc241650644editdlrm
caffe_translator_test.cpython-36.pyc29290644editdlrm
checkpoint.cpython-36.pyc279730644editdlrm
checkpoint_test.cpython-36.pyc94620644editdlrm
cnn.cpython-36.pyc84350644editdlrm
context.cpython-36.pyc40080644editdlrm
context_test.cpython-36.pyc26650644editdlrm
control.cpython-36.pyc144660644editdlrm
control_ops_grad.cpython-36.pyc163650644editdlrm
control_ops_grad_test.cpython-36.pyc15020644editdlrm
control_ops_util.cpython-36.pyc83320644editdlrm
control_test.cpython-36.pyc118700644editdlrm
convert.cpython-36.pyc1480644editdlrm
convert_test.cpython-36.pyc5680644editdlrm
convnet_benchmarks.cpython-36.pyc129450644editdlrm
convnet_benchmarks_test.cpython-36.pyc10900644editdlrm
core.cpython-36.pyc944280644editdlrm
core_gradients_test.cpython-36.pyc248250644editdlrm
core_test.cpython-36.pyc339420644editdlrm
crf.cpython-36.pyc76220644editdlrm
crf_predict.cpython-36.pyc10440644editdlrm
crf_viterbi_test.cpython-36.pyc16770644editdlrm
dataio.cpython-36.pyc239780644editdlrm
dataio_test.cpython-36.pyc132670644editdlrm
dataset.cpython-36.pyc128790644editdlrm
data_parallel_model.cpython-36.pyc514160644editdlrm
data_parallel_model_test.cpython-36.pyc389230644editdlrm
data_workers.cpython-36.pyc131420644editdlrm
data_workers_test.cpython-36.pyc45220644editdlrm
db_file_reader.cpython-36.pyc53610644editdlrm
db_test.cpython-36.pyc13280644editdlrm
device_checker.cpython-36.pyc40140644editdlrm
dyndep.cpython-36.pyc15680644editdlrm
embedding_generation_benchmark.cpython-36.pyc42030644editdlrm
experiment_util.cpython-36.pyc32330644editdlrm
extension_loader.cpython-36.pyc5430644editdlrm
fakefp16_transform_lib.cpython-36.pyc5460644editdlrm
filler_test.cpython-36.pyc9250644editdlrm
functional.cpython-36.pyc35310644editdlrm
functional_test.cpython-36.pyc41320644editdlrm
fused_8bit_rowwise_conversion_ops_test.cpython-36.pyc36310644editdlrm
gradient_checker.cpython-36.pyc106220644editdlrm
gradient_check_test.cpython-36.pyc167880644editdlrm
gru_cell.cpython-36.pyc25960644editdlrm
hip_test_util.cpython-36.pyc6690644editdlrm
hsm_util.cpython-36.pyc18230644editdlrm
hypothesis_test.cpython-36.pyc843120644editdlrm
hypothesis_test_util.cpython-36.pyc200390644editdlrm
ideep_test_util.cpython-36.pyc10810644editdlrm
layers_test.cpython-36.pyc573720644editdlrm
layer_model_helper.cpython-36.pyc216410644editdlrm
layer_model_instantiator.cpython-36.pyc36650644editdlrm
layer_parameter_sharing_test.cpython-36.pyc56260644editdlrm
layer_test_util.cpython-36.pyc52730644editdlrm
lazy.cpython-36.pyc4240644editdlrm
lazy_dyndep.cpython-36.pyc24750644editdlrm
lazy_dyndep_test.cpython-36.pyc48070644editdlrm
lengths_reducer_fused_8bit_rowwise_ops_test.cpython-36.pyc43920644editdlrm
lengths_reducer_rowwise_8bit_ops_test.cpython-36.pyc36840644editdlrm
lstm_benchmark.cpython-36.pyc72820644editdlrm
memonger.cpython-36.pyc321700644editdlrm
memonger_test.cpython-36.pyc231000644editdlrm
mkl_test_util.cpython-36.pyc11910644editdlrm
model_device_test.cpython-36.pyc33760644editdlrm
model_helper.cpython-36.pyc186180644editdlrm
model_helper_test.cpython-36.pyc19910644editdlrm
modifier_context.cpython-36.pyc26500644editdlrm
muji.cpython-36.pyc55660644editdlrm
muji_test.cpython-36.pyc35670644editdlrm
net_builder.cpython-36.pyc267860644editdlrm
net_builder_test.cpython-36.pyc92550644editdlrm
net_drawer.cpython-36.pyc100560644editdlrm
net_printer.cpython-36.pyc143670644editdlrm
net_printer_test.cpython-36.pyc33230644editdlrm
nomnigraph.cpython-36.pyc51250644editdlrm
nomnigraph_test.cpython-36.pyc145910644editdlrm
nomnigraph_transformations.cpython-36.pyc24310644editdlrm
nomnigraph_transformations_test.cpython-36.pyc40730644editdlrm
normalizer.cpython-36.pyc19410644editdlrm
normalizer_context.cpython-36.pyc15530644editdlrm
normalizer_test.cpython-36.pyc8700644editdlrm
numa_benchmark.cpython-36.pyc18860644editdlrm
numa_test.cpython-36.pyc16260644editdlrm
observer_test.cpython-36.pyc41810644editdlrm
operator_fp_exceptions_test.cpython-36.pyc13050644editdlrm
optimizer.cpython-36.pyc456060644editdlrm
optimizer_context.cpython-36.pyc20080644editdlrm
optimizer_test.cpython-36.pyc247470644editdlrm
optimizer_test_util.cpython-36.pyc68080644editdlrm
parallelize_bmuf_distributed_test.cpython-36.pyc69900644editdlrm
parallel_workers.cpython-36.pyc91480644editdlrm
parallel_workers_test.cpython-36.pyc38240644editdlrm
pipeline.cpython-36.pyc129400644editdlrm
pipeline_test.cpython-36.pyc27150644editdlrm
predictor_constants.cpython-36.pyc3040644editdlrm
python_op_test.cpython-36.pyc105020644editdlrm
queue_util.cpython-36.pyc50550644editdlrm
record_queue.cpython-36.pyc42250644editdlrm
recurrent.cpython-36.pyc98930644editdlrm
regularizer.cpython-36.pyc186040644editdlrm
regularizer_context.cpython-36.pyc15620644editdlrm
regularizer_test.cpython-36.pyc88250644editdlrm
rnn_cell.cpython-36.pyc447850644editdlrm
schema.cpython-36.pyc417630644editdlrm
schema_test.cpython-36.pyc137870644editdlrm
scope.cpython-36.pyc26000644editdlrm
scope_test.cpython-36.pyc40680644editdlrm
session.cpython-36.pyc73460644editdlrm
session_test.cpython-36.pyc23530644editdlrm
sparse_to_dense_mask_test.cpython-36.pyc52980644editdlrm
sparse_to_dense_test.cpython-36.pyc30030644editdlrm
task.cpython-36.pyc222070644editdlrm
task_test.cpython-36.pyc11780644editdlrm
test_util.cpython-36.pyc37290644editdlrm
text_file_reader.cpython-36.pyc22390644editdlrm
timeout_guard.cpython-36.pyc31060644editdlrm
toy_regression_test.cpython-36.pyc23600644editdlrm
transformations.cpython-36.pyc18330644editdlrm
transformations_test.cpython-36.pyc96080644editdlrm
tt_core.cpython-36.pyc64680644editdlrm
tt_core_test.cpython-36.pyc18240644editdlrm
utils.cpython-36.pyc122450644editdlrm
utils_test.cpython-36.pyc14160644editdlrm
visualize.cpython-36.pyc60270644editdlrm
workspace.cpython-36.pyc226900644editdlrm
workspace_test.cpython-36.pyc290550644editdlrm
_import_c_extension.cpython-36.pyc15240644editdlrm
__init__.cpython-36.pyc26070644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/caffe2/python/__pycache__/recurrent.cpython-36.pyc (9893B)
3 Eg3@sBddlmZmZddlmZmZd ddZdddZdd d ZdS))core workspace) viewitemsviewkeysNFc :s6t|dkstddd|D} dd|D} |jdfdd} d d| | Dd d|D7|d k rjt|fd djjD} tjj}d d|D}| st j j jj |\}ddt Dt jd}|jj d d =|d k rhdd|D}xNjj D]@}|jt|js$|jj j|gt|jj|s$tq$W|jj j|t j|j\}}t j|t jj\}}fddt |D}|jjj|d|j_nd }dd|Ddd|D| g}dj_g}g}g}g}x|D] \}}t|}| |d}|d} |t|}!|j||df|j|!|df|j||!ddf|j||!dd1f|j|!d|!dg|j||d k r$|j|!d| df|jjjt|!d|d}"|j|"ds |j|| dfn|j|"| dfq$Wx*|D]"\}#}$|jt|#t|$dfq/ If not provided we generate a scope name automatically outputs_with_grads : position indices of output blobs which will receive error gradient (from outside recurrent network) during backpropagation recompute_blobs_on_backward: specify a list of blobs that will be recomputed for backward pass, and thus need not to be stored for each forward timestep. forward_only: if True, only forward steps are executed z'Only one input blob is supported so farcSsg|]}t|dqS)r)str).0ir C/usr/local/lib64/python3.6/site-packages/caffe2/python/recurrent.py 3sz!recurrent_net..cSsg|]}t|dqS)r)r)rxr r r r 4sZ recurrentcs$dkr n}djt|t|S)Nz{}/{})formatr)nameZ scope_name)op_namescoper r s7szrecurrent_net..scSsg|] }t|qSr )r)rbr r r r ?scSsg|]}t|dqS)r)r)rr r r r r @sNcsg|]}|krtj|qSr )r BlobReference)rr) known_inputsr r r DscSsi|]}|d|qS)_gradr )ror r r Isz!recurrent_net..cSsi|]\}}|t|qSr )r)rkvr r r rOsZRecurrentBackwardStepcSsh|] }t|qSr )r)rrr r r Xsz recurrent_net..cs2g|]*\}}|dkr|kr|jjkr|qS)r)Protoexternal_output)rZblobver)cell_net undefinedr r r lssimplecSsg|] }|dqS)rr )rr r r r r wscSsg|] }|dqS)rr )rr r r r r xs_statesrrZ_all_lastcSs&|rt|\}}}|||fSgggfS)N)zip)r arcr r r unpack_triples z$recurrent_net..unpack_triplecSsg|]}t|dqS)r)r)rr r r r r sZ_accumcSs,ddt|jDddt|jDS)NcSsg|] }t|qSr )r)rr r r r r sz;recurrent_net..map_to_dual_list..cSsg|] }t|qSr )r)rr r r r r s)listkeysvalues)mr r r map_to_dual_listsz'recurrent_net..map_to_dual_listcsg|]}|kr|qSr r )rr )backward_mapping_keysr r r scs g|]}|krt|qSr )r)rr )backward_mappingr-r r r scsg|]}j|qSr )index)rp) all_inputsr r r scSsg|] }t|qSr )r)rlr r r r scSsg|] }t|qSr )r)rr2r r r r scSsg|] }t|qSr )r)rrr r r r s)parambackward_link_internalbackward_link_externalbackward_link_offsetoutputs_with_gradsrecompute_blobs_on_backward param_gradsZbackward_step_netZstep_workspacescSsg|] }t|qSr )r)rr%r r r r scsg|]}j|qSr )r/)rr )r1r r r scSsg|] }t|qSr )r)rr2r r r r scSsg|] }t|qSr )r)rr2r r r r stimestep) alias_src alias_dst alias_offsetrecurrent_statesZinitial_recurrent_state_ids link_internal link_external link_offsetZenable_rnn_executorZstep_netr:rBrBrBrB)lenAssertionErrorZNextNameappendrrZexternal_inputr(rrZGradientRegistryZGetBackwardPassoprZNet isdisjointsetoutputextendissubsetZget_ssaZget_undefined_blobstypegetremove enumerateinputrZSumrRecurrentNetwork):netrinputsZinitial_cell_inputslinksr:rr7r8Z forward_onlyZ input_blobsZinitial_input_blobsrZ referencesZ inner_outputsZinner_outputs_mapZ backward_opsZbackward_cell_netrFZ backward_ssaZbackward_blob_versionsZssaZ blob_versionsZ scratchesZ all_outputsZ forward_linksZbackward_linksaliasesr>Z cell_input_stateZ states_gradZ cell_outputZrecurrent_input_gradZinput_tZ input_blobr'r?r@rAr;r<r=Zrecurrent_inputsproto operatorsjZ output_blobZ accum_blobr,Z backward_argsr4r5r6paramsr9resultsr )r1r.r-rrrrr r recurrent_net s&               "                      r]csPddlmjd kstfdd}|dk r:|d||dk rL|d|dS) Nr) caffe2_pb2rQRecurrentNetworkGradientcs*j}d||_||_jj|gdS)Nz rnn_executor.)ZArgumentrr argrJ)rrr%)r^rnn_opr r add_arg%s z(set_rnn_executor_config..add_arg num_threadsmax_cuda_streams>r_rQ)Z caffe2.protor^rLrD)rarcrdrbr )r^rar set_rnn_executor_config!s  rernncCszd}g}xl|jjD]^}|jdkr|dt|}|d}|jd}tjtjd|g|g|d|tj |j 7}qW|S)a Retrieves blobs from step workspaces (which contain intermediate recurrent network computation for each timestep) and puts them in the global workspace. This allows access to the contents of this intermediate computation in python. Returns the list of extracted blob names. net: the net from which the step workspace blobs should be extracted prefix: prefix to append to extracted blob names when placing them in the global workspace rrQrVZRecurrentNetworkBlobFetcher)prefixrB) rrFrLrrIrZRunOperatorOncerZCreateOperatorZ FetchBlobtolist)rRrgcountZ output_listrFZ blob_nameZscratch_workspaces_blob_namer r r retrieve_step_blobs1s    rjr)NNrkNF)NN)rf) Z caffe2.pythonrrZ future.utilsrrr]rerjr r r r s