| /external/tensorflow/tensorflow/python/ops/ |
| D | map_fn.py | 53 infer_shape=True, argument
450 dtype=t.dtype, size=n, dynamic_size=False, infer_shape=True)
469 infer_shape=infer_shape, element_shape=spec.shape))
637 infer_shape=True, argument
650 infer_shape=infer_shape,
|
| D | tensor_array_ops.py | 64 infer_shape=True, argument
141 self._infer_shape = infer_shape
159 identical_element_shapes=infer_shape,
246 infer_shape=self._infer_shape,
405 infer_shape=True, argument
461 self._infer_shape = infer_shape
662 infer_shape=True, argument
706 self._infer_shape = infer_shape
1010 infer_shape=True, argument
1076 infer_shape=infer_shape,
[all …]
|
| D | tensor_array_ops_test.py | 37 size=4, dtype=dtypes.string, element_shape=[None], infer_shape=False)
55 size=4, dtype=dtypes.string, element_shape=[None], infer_shape=False)
68 size=4, dtype=dtypes.string, element_shape=[None], infer_shape=False)
|
| D | functional_ops.py | 111 infer_shape=True).unstack(elem)
306 infer_shape=True).unstack(elem)
447 infer_shape=True, argument
598 infer_shape=True) for elem in elems_flat
618 element_shape=init.shape if infer_shape else None,
620 infer_shape=infer_shape) for init in a_flat
711 infer_shape=True, argument
813 infer_shape=infer_shape,
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_map_fn_op_test.py | 180 infer_shape=True, argument
184 fn=fn, elems=elems, dtype=result_dtype, infer_shape=infer_shape)
225 infer_shape=False,
252 infer_shape=False)
273 infer_shape=False)
|
| D | ragged_map_ops.py | 36 infer_shape=True, argument
170 infer_shape,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tensor_array_ops_test.py | 65 def _make_ta(size, name, dtype=dtypes.float32, infer_shape=False): argument
67 dtype=dtype, tensor_array_name=name, size=size, infer_shape=infer_shape)
91 infer_shape=False)
155 dtype=dtypes.int32, size=2, infer_shape=False)
164 dtype=dtypes.int32, size=3, infer_shape=False)
182 infer_shape=False))
194 dtype=tf_dtype, tensor_array_name="foo", size=3, infer_shape=False)
373 infer_shape=False)
411 infer_shape=False)
429 infer_shape=False)
[all …]
|
| D | garbage_collection_test.py | 46 infer_shape=False)
|
| D | functional_ops_test.py | 398 result_t = functional_ops.scan(lambda a, x: a + x, x_t, infer_shape=False)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tensor_array_ops_test.py | 446 infer_shape=False)
458 infer_shape=False)
470 infer_shape=False)
482 infer_shape=False)
497 dtype=dtype, tensor_array_name="foo", size=3, infer_shape=False)
515 dtype=dtype, tensor_array_name="foo", size=3, infer_shape=False)
563 infer_shape=False)
759 infer_shape=False)
984 dtype=dtypes.float32, size=0, infer_shape=False)
1001 dtype=dtypes.float32, size=0, infer_shape=True)
[all …]
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure.py | 56 def _TensorArrayStructure(dtype, element_shape, dynamic_size, infer_shape): argument
58 dynamic_size, infer_shape)
178 infer_shape=tensor_shape.dimension_value(flat_shape[1])))
|
| D | structure_test.py | 558 [2, 2], dtypes.int32, dynamic_size=None, infer_shape=True)),
563 [2, 2], dtypes.int32, dynamic_size=True, infer_shape=None)),
568 [2, 2], dtypes.int32, dynamic_size=True, infer_shape=False)),
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-tensor-array-spec.pbtxt | 12 …argspec: "args=[\'self\', \'element_shape\', \'dtype\', \'dynamic_size\', \'infer_shape\'], vararg…
|
| D | tensorflow.-tensor-array.pbtxt | 27 …, \'clear_after_read\', \'tensor_array_name\', \'handle\', \'flow\', \'infer_shape\', \'element_sh…
|
| D | tensorflow.data.experimental.pbtxt | 105 …argspec: "args=[\'dtype\', \'element_shape\', \'dynamic_size\', \'infer_shape\'], varargs=None, ke…
|
| D | tensorflow.pbtxt | 1629 … \'dtype\', \'parallel_iterations\', \'back_prop\', \'swap_memory\', \'infer_shape\', \'name\', \'…
1997 …tializer\', \'parallel_iterations\', \'back_prop\', \'swap_memory\', \'infer_shape\', \'reverse\',…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-tensor-array-spec.pbtxt | 12 …argspec: "args=[\'self\', \'element_shape\', \'dtype\', \'dynamic_size\', \'infer_shape\'], vararg…
|
| D | tensorflow.-tensor-array.pbtxt | 27 …, \'clear_after_read\', \'tensor_array_name\', \'handle\', \'flow\', \'infer_shape\', \'element_sh…
|
| D | tensorflow.pbtxt | 797 … \'dtype\', \'parallel_iterations\', \'back_prop\', \'swap_memory\', \'infer_shape\', \'name\', \'…
977 …tializer\', \'parallel_iterations\', \'back_prop\', \'swap_memory\', \'infer_shape\', \'reverse\',…
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | data_structures.py | 100 infer_shape=(element_shape is None),
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | subscribe_test.py | 264 infer_shape=False)
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 4347 infer_shape=True))
|