Searched refs:captured_tensor (Results 1 – 7 of 7) sorted by relevance
1133 captured_tensor = self._indirect_captures.get(ops.tensor_id(tensor))1134 if captured_tensor is not None:1135 return captured_tensor1139 captured_tensor = super(_WhileBodyGradFuncGraph, self)._capture_helper(1145 captured_tensor)] = captured_tensor1146 self._indirect_captures[ops.tensor_id(tensor)] = captured_tensor1147 return captured_tensor1154 captured_tensor = self._indirect_captures.get(ops.tensor_id(tensor))1155 if captured_tensor is not None:1156 return captured_tensor[all …]
976 captured_tensor = self._indirect_captures.get(tensor_id)977 if captured_tensor is not None:978 return captured_tensor995 captured_tensor = super(_CondGradFuncGraph, self)._capture_helper(1017 captured_tensor = gen_dataset_ops.optional_get_value(1020 self._indirect_captures[tensor_id] = captured_tensor1021 return captured_tensor
628 self._get_tensor_from_fn, proto.captured_tensor),714 captured_tensor = outer_graph.get_tensor_by_name(proto.name)715 return captured_tensor, setattr
994 proto.captured_tensor.name = obj.name995 proto.captured_tensor.concrete_function = obj.concrete_function
42 // "constant" and "captured_tensor".71 CapturedTensor captured_tensor = 12; field
307 for (const Tensor& captured_tensor : captured_tensors) { in RegisterInput() local308 batch_components->captured_inputs.push_back(captured_tensor); in RegisterInput()
5304 captured_tensor = constant_op.constant([1.])5312 return deferred_tensor + captured_tensor