Searched refs:shape_invariants (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops.py | 2877 shape_invariants): argument
2894 use_input_shape=(shape_invariants is None)) for x in real_vars
2916 _SetShapeInvariants(real_vars, enter_vars, shape_invariants)
2995 def BuildLoop(self, pred, body, loop_vars, shape_invariants): argument
3007 pred, body, original_loop_vars, loop_vars, shape_invariants)
3054 shape_invariants=None, argument
3255 if shape_invariants is not None:
3257 shape_invariants = (tensor_shape.TensorShape([]), shape_invariants)
3258 nest.assert_same_structure(loop_vars, shape_invariants)
3272 result = loop_context.BuildLoop(cond, body, loop_vars, shape_invariants)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_tensor_dense_matmul_op_test.py | 269 shape_invariants=(tensor_shape.TensorShape(()),
296 shape_invariants=(tensor_shape.TensorShape(()),
|
| D | tensor_array_ops_test.py | 854 shape_invariants=(time_0.get_shape(), tensor_shape.unknown_shape(),
|
| /external/tensorflow/tensorflow/contrib/losses/python/metric_learning/ |
| D | metric_loss_ops.py | 763 shape_invariants=[iteration.get_shape(), tensor_shape.TensorShape(
|
| /external/tensorflow/tensorflow/tools/api/golden/ |
| D | tensorflow.pbtxt | 2101 …argspec: "args=[\'cond\', \'body\', \'loop_vars\', \'shape_invariants\', \'parallel_iterations\', …
|