Searched refs:ensure_shape (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | ensure_shape_op_test.py | 35 op = check_ops.ensure_shape(p, (None, 3))
44 op = check_ops.ensure_shape(p, (None, 3, 3))
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | check_ops_test.py | 932 ensure_shape_op = check_ops.ensure_shape(placeholder, (3, 3, 3))
938 ensure_shape_op = check_ops.ensure_shape(placeholder, (5, 4, None))
945 check_ops.ensure_shape(placeholder, (2, 3))
951 check_ops.ensure_shape(placeholder, (2, 2, 4))
957 ensure_shape_op = check_ops.ensure_shape(derived, None)
967 derived = check_ops.ensure_shape(derived, (3, 3, 3))
982 derived = check_ops.ensure_shape(derived, (None, None, 3))
995 derived = check_ops.ensure_shape(derived, (2, 1))
1004 derived = check_ops.ensure_shape(derived, (None, None))
1012 derived = check_ops.ensure_shape(placeholder, (None, None))
[all …]
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scaling_gradient_tape.py | 294 g = array_ops.ensure_shape(g, initial_grad.shape)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | check_ops.py | 2222 def ensure_shape(x, shape, name=None): function
2361 return array_ops.ensure_shape(x, shape, name=name)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | map_vectorization_test.py | 576 return check_ops.ensure_shape(
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 628 name: "ensure_shape"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 1224 name: "ensure_shape"
|