↓ 3 callersMethodfind_inf(self, sizea, sizeb, applier, repeat_tensors, in_type, out_type, t, ind, val, inplace=False)
segmentation/apex/tests/L0/run_amp/test_multi_tensor_scale.py:55
↓ 3 callersFunctionget_loader(model, image_path, metadata_path, mode, batch_size, is_shuffle=False, num_val=100)
test/posenet/data_loader.py:81
↓ 2 callersMethod__init__(self, gate_multiplier, input_size, hidden_size, cell, n_hidden_states = 2, bias = False, output_size = None)
segmentation/apex/apex/RNN/RNNBackend.py:242
↓ 2 callersMethod__launch_step_kernel(self, p, p_copy, m, v, g)
segmentation/apex/apex/contrib/optimizers/distributed_fused_adam_v2.py:406
↓ 2 callersMethodaxpby(self, sizea, sizeb, applier, repeat_tensors,
x_type, y_type, out_type, inplace=False, nhwc=Fals
segmentation/apex/tests/L0/run_amp/test_multi_tensor_axpby.py:45