Searched refs:out_tensor_impl (Results 1 – 1 of 1) sorted by relevance
117 auto out_tensor_impl = raw_outputs[i].value().unsafeGetTensorImpl(); in _process_forward_mode_AD() local119 (non_differentiable.count(out_tensor_impl) == 0 && in _process_forward_mode_AD()133 bool is_input = inputs_mapping.count(out_tensor_impl) > 0; in _process_forward_mode_AD()134 bool is_modified = dirty_inputs.count(out_tensor_impl) > 0; in _process_forward_mode_AD()141 auto inp_idx = inputs_mapping[out_tensor_impl]; in _process_forward_mode_AD()368 auto out_tensor_impl = var.unsafeGetTensorImpl(); in _process_backward_mode_ad() local369 bool is_input = inputs_mapping.count(out_tensor_impl) > 0; in _process_backward_mode_ad()370 bool is_modified = dirty_inputs.count(out_tensor_impl) > 0; in _process_backward_mode_ad()372 non_differentiable.count(out_tensor_impl) == 0 && in _process_backward_mode_ad()375 to_save_if_setup_context.count(out_tensor_impl) > 0; in _process_backward_mode_ad()[all …]