Searched refs:shape_invariants (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2.py | 59 shape_invariants=None, argument
76 if shape_invariants is not None:
77 nest.assert_same_structure(orig_loop_vars, shape_invariants)
79 shape_invariants = nest.map_structure(lambda t: t.shape, loop_vars)
98 shape_invariants = type(shape_invariants)(
99 [tensor_shape.scalar(), tensor_shape.scalar()]) + shape_invariants
126 signature=_build_signature(loop_vars, shape_invariants),
170 signature=_build_signature(loop_vars, shape_invariants),
203 shape_invariants[first_loop_var_index:first_loop_var_index +
904 def _check_shapes_compat(output_tensors, shape_invariants, input_tensors): argument
[all …]
|
| D | control_flow_ops.py | 2851 shape_invariants): argument
2868 use_input_shape=(shape_invariants is None)) for x in real_vars
2890 _SetShapeInvariants(real_vars, enter_vars, shape_invariants)
2975 def BuildLoop(self, pred, body, loop_vars, shape_invariants, argument
2987 if shape_invariants is None:
2988 shape_invariants = nest.map_structure(
2998 pred, body, original_loop_vars, loop_vars, shape_invariants)
3045 shape_invariants=None, argument
3210 shape_invariants=shape_invariants,
3224 shape_invariants=None, argument
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_tensor_dense_matmul_op_test.py | 276 shape_invariants=(tensor_shape.TensorShape(()),
303 shape_invariants=(tensor_shape.TensorShape(()),
|
| D | tensor_array_ops_test.py | 946 shape_invariants=(time_0.get_shape(), tensor_shape.unknown_shape(),
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 436 shape_invariants = (
441 return init_values, inputs_stacked, shape_invariants
584 init_values, inputs_stacked, shape_invariants = self._create_init_values(
644 cond, body, init_values, shape_invariants=shape_invariants)
|
| /external/tensorflow/tensorflow/contrib/losses/python/metric_learning/ |
| D | metric_loss_ops.py | 769 shape_invariants=[iteration.get_shape(), tensor_shape.TensorShape(
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | importer_test.py | 459 shape_invariants=[tensor_shape.TensorShape(None)])
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 1073 …argspec: "args=[\'cond\', \'body\', \'loop_vars\', \'shape_invariants\', \'parallel_iterations\', …
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 2421 …argspec: "args=[\'cond\', \'body\', \'loop_vars\', \'shape_invariants\', \'parallel_iterations\', …
|