| /external/tensorflow/tensorflow/python/autograph/tests/ |
| D | loop_with_variable_type_test.py | 31 shape_invariants=[(v, tf.TensorShape([None]))])
41 shape_invariants=[(v, tf.TensorShape([None]))])
51 shape_invariants=[(m, tf.TensorShape([None, 1]))])
61 shape_invariants=[(m, tf.TensorShape([None, 1]))])
71 shape_invariants=[(m, tf.TensorShape([1, None]))])
81 shape_invariants=[(m, tf.TensorShape([1, None]))])
91 shape_invariants=[(m, tf.TensorShape(None))])
101 shape_invariants=[(m, tf.TensorShape(None))])
112 shape_invariants=[(v, tf.TensorShape([None]))])
125 shape_invariants=[(v, tf.TensorShape([None]))])
[all …]
|
| D | loop_with_variable_type_illegal_cases_test.py | 95 shape_invariants=((t, tf.TensorShape([1])),))
104 shape_invariants=((t, tf.TensorShape([1])),))
|
| /external/tensorflow/tensorflow/python/autograph/lang/ |
| D | directives.py | 50 shape_invariants=UNSPECIFIED): argument
94 del shape_invariants
|
| /external/tensorflow/tensorflow/python/ops/risc/ |
| D | risc_ops.py | 267 shape_invariants=None, argument
277 shape_invariants=shape_invariants,
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow.py | 274 shape_invariants = opts['shape_invariants']
276 shape_invariants = nest.map_structure(lambda _: None, iter_entry_vars)
278 assert len(symbol_names) == len(shape_invariants)
288 invariant = shape_invariants[i]
1123 shape_invariants, symbol_names): argument
1183 shape_invariants))
1231 shape_invariants = tuple(
1236 nulls, shape_invariants,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.autograph.experimental.pbtxt | 13 …\'parallel_iterations\', \'swap_memory\', \'maximum_iterations\', \'shape_invariants\'], varargs=N…
|
| D | tensorflow.pbtxt | 1177 …argspec: "args=[\'cond\', \'body\', \'loop_vars\', \'shape_invariants\', \'parallel_iterations\', …
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.autograph.experimental.pbtxt | 13 …\'parallel_iterations\', \'swap_memory\', \'maximum_iterations\', \'shape_invariants\'], varargs=N…
|
| D | tensorflow.pbtxt | 2541 …argspec: "args=[\'cond\', \'body\', \'loop_vars\', \'shape_invariants\', \'parallel_iterations\', …
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops.py | 2249 def BuildLoop(self, pred, body, loop_vars, shape_invariants, argument
2263 if shape_invariants is not None:
2265 _shape_invariant_to_type_spec, loop_vars, shape_invariants)
2350 shape_invariants=None, argument
2517 shape_invariants=shape_invariants,
2531 shape_invariants=None, argument
2717 shape_invariants=shape_invariants,
2779 if shape_invariants is not None:
2781 shape_invariants = (tensor_shape.TensorShape([]), shape_invariants)
2792 result = loop_context.BuildLoop(cond, body, loop_vars, shape_invariants,
|
| D | while_v2.py | 74 shape_invariants=None, argument
100 if shape_invariants is not None:
103 loop_vars, shape_invariants)
|
| D | image_ops_impl.py | 5637 shape_invariants=[
|
| /external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| D | sparse_tensor_dense_matmul_op_test.py | 279 shape_invariants=(tensor_shape.TensorShape(()),
306 shape_invariants=(tensor_shape.TensorShape(()),
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | constraints.py | 303 shape_invariants=[index.get_shape(),
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 556 shape_invariants = (
561 return init_values, inputs_stacked, shape_invariants
701 init_values, inputs_stacked, shape_invariants = self._create_init_values(
764 cond, body, init_values, shape_invariants=shape_invariants)
5016 shape_invariants = (
5022 shape_invariants=shape_invariants,
|
| D | control_flow_ops_test.py | 2003 shape_invariants=[
|
| /external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
| D | control_flow_ops_py_test.py | 2320 shape_invariants = [
2323 control_flow_ops.while_loop(c, b, [i, x], shape_invariants)
2332 shape_invariants = [
2335 control_flow_ops.while_loop(c, b, [i, x], shape_invariants)
2339 control_flow_ops.while_loop(c, b, [i, x2], shape_invariants)
2343 control_flow_ops.while_loop(c, b, [i, x3], shape_invariants)
2347 control_flow_ops.while_loop(c, b, [i2, x], shape_invariants)
4673 def _buildWhileWithShapeInvariants(self, shape_invariants): argument
4683 cond, body, [r], shape_invariants=shape_invariants)
|
| D | while_v2_test.py | 2055 shape_invariants=[
|
| /external/tensorflow/tensorflow/python/autograph/g3doc/reference/ |
| D | control_flow.md | 240 for TensorFlow entities, or declared in `shape_invariants`
|
| D | limitations.md | 667 do need to change shape across iterations, use `shape_invariants`.
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | importer_test.py | 456 shape_invariants=[tensor_shape.TensorShape(None)])
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training.py | 1586 shape_invariants=[(
|
| /external/tensorflow/tensorflow/python/kernel_tests/data_structures/ |
| D | tensor_array_ops_test.py | 993 shape_invariants=(time_0.get_shape(), tensor_shape.unknown_shape(),
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function_test.py | 5410 shape_invariants=[(sums, tensor_shape.TensorShape(None))])
|