| /external/tensorflow/tensorflow/python/ops/ragged/ |
| ragged_map_fn_op_test.py | 163 infer_shape=False,
167 fn=fn, elems=elems, dtype=result_dtype, infer_shape=infer_shape)
208 infer_shape=False,
235 infer_shape=False)
256 infer_shape=False)
|
| ragged_map_ops.py | 43 infer_shape=True,
121 infer_shape: (optional) False disables tests for consistent output shapes.
251 dtype=t, dynamic_size=False, infer_shape=infer_shape, size=n)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| garbage_collection_test.py | 46 infer_shape=False)
|
| tensor_array_ops_test.py | 62 def _make_ta(size, name, dtype=dtypes.float32, infer_shape=False):
64 dtype=dtype, tensor_array_name=name, size=size, infer_shape=infer_shape)
88 infer_shape=False)
150 dtype=tf_dtype, tensor_array_name="foo", size=3, infer_shape=False)
329 infer_shape=False)
367 infer_shape=False)
385 infer_shape=False)
521 infer_shape=False)
535 infer_shape=False
[all...] |
| functional_ops_test.py | 397 result_t = functional_ops.scan(lambda a, x: a + x, x_t, infer_shape=False)
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| map_fn.py | 39 swap_memory=False, infer_shape=True, name=None):
115 infer_shape: (optional) False disables tests for consistent output shapes.
227 infer_shape=True)
239 infer_shape=infer_shape)
|
| tensor_array_ops.py | 57 infer_shape=True,
79 infer_shape: (optional, default: True) If True, shape inference
133 # of the first write. If `infer_shape` is true, all writes checks for
136 self._infer_shape = infer_shape
157 identical_element_shapes=infer_shape,
195 "(and infer_shape=True)" % (shape, self._element_shape[0]))
228 infer_shape=self._infer_shape,
252 infer_shape=self._infer_shape,
426 infer_shape=True,
442 infer_shape: (optional, default: True) If True, shape inference i
[all...] |
| functional_ops.py | 102 infer_shape=True).unstack(elem)
211 infer_shape=True).unstack(elem)
268 swap_memory=False, infer_shape=True, reverse=False, name=None):
318 infer_shape: (optional) False disables tests for consistent output shapes.
409 infer_shape=True)
427 element_shape=init.shape if infer_shape else None,
429 infer_shape=infer_shape)
[all...] |
| control_flow_ops.py | 434 infer_shape=ta._infer_shape,
[all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| tensor_array_ops_test.py | 369 infer_shape=False)
388 infer_shape=False)
398 dtype=dtype, tensor_array_name="foo", size=3, infer_shape=False)
448 infer_shape=False)
620 infer_shape=False)
846 infer_shape=True)
871 infer_shape=True)
881 infer_shape=True)
892 infer_shape=True)
[all...] |
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| resample.py | 57 dtype=dtypes.int32, size=size, infer_shape=False)
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| data_structures.py | 87 # TODO(mdan): We may want to allow different shapes with infer_shape=False.
100 infer_shape=(element_shape is None),
|
| /external/tensorflow/tensorflow/python/framework/ |
| subscribe_test.py | 264 infer_shape=False)
|
| /external/tensorflow/tensorflow/python/grappler/ |
| hierarchical_controller.py | [all...] |
