Home
last modified time | relevance | path

Searched refs:input_ops (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dinput_ops_test.py22 from tensorflow.python.distribute import input_ops
111 dataset = input_ops.auto_shard_dataset(
121 dataset = input_ops.auto_shard_dataset(
132 dataset = input_ops.auto_shard_dataset(
145 dataset = input_ops.auto_shard_dataset(
175 dataset = input_ops.auto_shard_dataset(
202 dataset = input_ops.auto_shard_dataset(
214 dataset = input_ops.auto_shard_dataset(
233 dataset = input_ops.auto_shard_dataset(
241 dataset = input_ops.auto_shard_dataset(
[all …]
Dinput_lib.py34 from tensorflow.python.distribute import input_ops
1144 dataset = input_ops.auto_shard_dataset(dataset,
1157 cloned_dataset = input_ops.auto_shard_dataset(
DBUILD540 ":input_ops",
575 name = "input_ops",
576 srcs = ["input_ops.py"],
589 ":input_ops",
/external/tensorflow/tensorflow/python/ops/
Dcustom_gradient.py374 def _get_dependent_variables(input_ops, output_ops): argument
389 stop_at_ts=input_ops,
465 input_ops=filtered_input_tensors, output_ops=flat_result)
/external/tensorflow/tensorflow/python/training/
Dserver_lib_test.py34 from tensorflow.python.training import input as input_ops unknown
292 input_queue = input_ops.input_producer(constant_op.constant(
/external/tensorflow/tensorflow/python/tpu/
Dtensor_tracer.py1250 input_ops = [tensor_input.op for tensor_input in head_op.inputs]
1251 input_ops.extend(head_op.control_inputs)
1253 for input_op in input_ops:
/external/tensorflow/tensorflow/python/framework/
Dops.py3793 input_ops = set(t.op for t in inputs)
3794 control_inputs = self._control_dependencies_for_inputs(input_ops)
4945 def _control_dependencies_for_inputs(self, input_ops): argument
4968 for op in input_ops:
4976 ret.extend(c for c in controller.control_inputs if c not in input_ops)
/external/tensorflow/tensorflow/python/eager/
Dfunction.py323 input_ops = set(arg.op for arg in inputs)
324 operations = [op for op in graph.get_operations() if op not in input_ops]