Home
last modified time | relevance | path

Searched refs:captured_tensor (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dwhile_v2.py1133 captured_tensor = self._indirect_captures.get(ops.tensor_id(tensor))
1134 if captured_tensor is not None:
1135 return captured_tensor
1139 captured_tensor = super(_WhileBodyGradFuncGraph, self)._capture_helper(
1145 captured_tensor)] = captured_tensor
1146 self._indirect_captures[ops.tensor_id(tensor)] = captured_tensor
1147 return captured_tensor
1154 captured_tensor = self._indirect_captures.get(ops.tensor_id(tensor))
1155 if captured_tensor is not None:
1156 return captured_tensor
[all …]
Dcond_v2.py976 captured_tensor = self._indirect_captures.get(tensor_id)
977 if captured_tensor is not None:
978 return captured_tensor
995 captured_tensor = super(_CondGradFuncGraph, self)._capture_helper(
1017 captured_tensor = gen_dataset_ops.optional_get_value(
1020 self._indirect_captures[tensor_id] = captured_tensor
1021 return captured_tensor
/external/tensorflow/tensorflow/python/saved_model/
Dload.py628 self._get_tensor_from_fn, proto.captured_tensor),
714 captured_tensor = outer_graph.get_tensor_by_name(proto.name)
715 return captured_tensor, setattr
Dsave.py994 proto.captured_tensor.name = obj.name
995 proto.captured_tensor.concrete_function = obj.concrete_function
/external/tensorflow/tensorflow/core/protobuf/
Dsaved_object_graph.proto42 // "constant" and "captured_tensor".
71 CapturedTensor captured_tensor = 12; field
/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_resource_base.cc307 for (const Tensor& captured_tensor : captured_tensors) { in RegisterInput() local
308 batch_components->captured_inputs.push_back(captured_tensor); in RegisterInput()
/external/tensorflow/tensorflow/python/eager/
Dfunction_test.py5304 captured_tensor = constant_op.constant([1.])
5312 return deferred_tensor + captured_tensor