Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/
Dinput_ops_test.py26 from tensorflow.python.distribute import input_ops
115 dataset = input_ops.auto_shard_dataset(
125 dataset = input_ops.auto_shard_dataset(
136 dataset = input_ops.auto_shard_dataset(
149 dataset = input_ops.auto_shard_dataset(
179 dataset = input_ops.auto_shard_dataset(
206 dataset = input_ops.auto_shard_dataset(
218 dataset = input_ops.auto_shard_dataset(
237 dataset = input_ops.auto_shard_dataset(
245 dataset = input_ops.auto_shard_dataset(
[all …]
Dinput_lib.py37 from tensorflow.python.distribute import input_ops
1014 dataset = input_ops.auto_shard_dataset(dataset,
1028 cloned_dataset = input_ops.auto_shard_dataset(
DBUILD514 ":input_ops",
531 name = "input_ops",
532 srcs = ["input_ops.py"],
545 ":input_ops",
/external/tensorflow/tensorflow/python/ops/
Dcustom_gradient.py287 def _get_dependent_variables(input_ops, output_ops): argument
302 stop_at_ts=input_ops,
374 input_ops=filtered_input_tensors, output_ops=flat_result)
/external/tensorflow/tensorflow/python/framework/
Derror_interpolation.py455 def _build_error_message(op, input_ops, common_prefix): argument
473 for op_inp in input_ops:
Dops.py3540 input_ops = set(t.op for t in inputs)
3541 control_inputs = self._control_dependencies_for_inputs(input_ops)
4686 def _control_dependencies_for_inputs(self, input_ops): argument
4709 for op in input_ops:
4717 ret.extend(c for c in controller.control_inputs if c not in input_ops)
/external/tensorflow/tensorflow/python/training/
Dserver_lib_test.py38 from tensorflow.python.training import input as input_ops unknown
296 input_queue = input_ops.input_producer(constant_op.constant(
/external/tensorflow/tensorflow/python/tpu/
Dtensor_tracer.py1165 input_ops = [tensor_input.op for tensor_input in head_op.inputs]
1166 input_ops.extend(head_op.control_inputs)
1168 for input_op in input_ops:
/external/tensorflow/tensorflow/python/eager/
Dfunction.py445 input_ops = set(arg.op for arg in inputs)
446 operations = [op for op in graph.get_operations() if op not in input_ops]