Searched refs:captured_tensor (Results 1 – 3 of 3) sorted by relevance
1084 captured_tensor = self._indirect_captures.get(ops.tensor_id(tensor))1085 if captured_tensor is not None:1086 return captured_tensor1090 captured_tensor = super(_WhileBodyGradFuncGraph, self)._capture_helper(1096 captured_tensor)] = captured_tensor1097 self._indirect_captures[ops.tensor_id(tensor)] = captured_tensor1098 return captured_tensor1105 captured_tensor = self._indirect_captures.get(ops.tensor_id(tensor))1106 if captured_tensor is not None:1107 return captured_tensor[all …]
935 captured_tensor = self._indirect_captures.get(tensor_id)936 if captured_tensor is not None:937 return captured_tensor954 captured_tensor = super(_CondGradFuncGraph, self)._capture_helper(976 captured_tensor = gen_dataset_ops.optional_get_value(979 self._indirect_captures[tensor_id] = captured_tensor980 return captured_tensor
210 for (const Tensor& captured_tensor : captured_tensors) { in RegisterInput() local211 batch_components->captured_inputs.push_back(captured_tensor); in RegisterInput()