Searched refs:captured_tensor (Results 1 – 4 of 4) sorted by relevance
650 captured_tensor = self._indirect_captures.get(tensor)651 if captured_tensor is not None:652 return captured_tensor678 captured_tensor = gen_dataset_ops.optional_get_value(680 self._indirect_captures[tensor] = captured_tensor681 return captured_tensor
751 captured_tensor = self._indirect_captures.get(tensor)752 if captured_tensor is not None:753 return captured_tensor791 new_tensor_list, captured_tensor = list_ops.tensor_list_pop_back(794 self._indirect_captures[tensor] = captured_tensor796 return captured_tensor
494 captured_tensor = self.captures.get(tensor, None)495 if captured_tensor is None:496 captured_tensor = _create_substitute_placeholder(tensor, name=name,498 self.captures[tensor] = captured_tensor499 self.inputs.append(captured_tensor)500 tape.record_operation("captured_value", [captured_tensor], [tensor],502 return captured_tensor
265 for (const Tensor& captured_tensor : captured_tensors) { in RegisterInput() local266 batch_components->captured_inputs.push_back(captured_tensor); in RegisterInput()