Home
last modified time | relevance | path

Searched refs:infer_shape (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_map_fn_op_test.py163 infer_shape=False, argument
167 fn=fn, elems=elems, dtype=result_dtype, infer_shape=infer_shape)
208 infer_shape=False,
235 infer_shape=False)
256 infer_shape=False)
Dragged_map_ops.py43 infer_shape=True, argument
251 dtype=t, dynamic_size=False, infer_shape=infer_shape, size=n)
/external/tensorflow/tensorflow/python/ops/
Dtensor_array_ops.py57 infer_shape=True, argument
136 self._infer_shape = infer_shape
157 identical_element_shapes=infer_shape,
228 infer_shape=self._infer_shape,
252 infer_shape=self._infer_shape,
426 infer_shape=True, argument
481 self._infer_shape = infer_shape
688 infer_shape=True, argument
732 self._infer_shape = infer_shape
975 infer_shape=True, argument
[all …]
Dmap_fn.py39 swap_memory=False, infer_shape=True, name=None): argument
227 infer_shape=True)
239 infer_shape=infer_shape)
Dfunctional_ops.py102 infer_shape=True).unstack(elem)
211 infer_shape=True).unstack(elem)
268 swap_memory=False, infer_shape=True, reverse=False, name=None): argument
409 infer_shape=True)
427 element_shape=init.shape if infer_shape else None,
429 infer_shape=infer_shape)
Dcontrol_flow_ops.py434 infer_shape=ta._infer_shape,
/external/tensorflow/tensorflow/python/kernel_tests/
Dtensor_array_ops_test.py62 def _make_ta(size, name, dtype=dtypes.float32, infer_shape=False): argument
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)
591 dtype=dtype, tensor_array_name="foo", size=3, infer_shape=False)
[all …]
Dgarbage_collection_test.py46 infer_shape=False)
Dfunctional_ops_test.py397 result_t = functional_ops.scan(lambda a, x: a + x, x_t, infer_shape=False)
/external/tensorflow/tensorflow/compiler/tests/
Dtensor_array_ops_test.py369 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)
915 dtype=dtypes.float32, size=0, infer_shape=False)
[all …]
/external/tensorflow/tensorflow/contrib/training/python/training/
Dresample.py57 dtype=dtypes.int32, size=size, infer_shape=False)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-tensor-array.pbtxt19 …, \'clear_after_read\', \'tensor_array_name\', \'handle\', \'flow\', \'infer_shape\', \'element_sh…
Dtensorflow.pbtxt1565 … \'dtype\', \'parallel_iterations\', \'back_prop\', \'swap_memory\', \'infer_shape\', \'name\'], v…
1913 …tializer\', \'parallel_iterations\', \'back_prop\', \'swap_memory\', \'infer_shape\', \'reverse\',…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-tensor-array.pbtxt19 …, \'clear_after_read\', \'tensor_array_name\', \'handle\', \'flow\', \'infer_shape\', \'element_sh…
Dtensorflow.pbtxt733 … \'dtype\', \'parallel_iterations\', \'back_prop\', \'swap_memory\', \'infer_shape\', \'name\'], v…
905 …tializer\', \'parallel_iterations\', \'back_prop\', \'swap_memory\', \'infer_shape\', \'reverse\',…
/external/tensorflow/tensorflow/python/autograph/operators/
Ddata_structures.py100 infer_shape=(element_shape is None),
/external/tensorflow/tensorflow/python/framework/
Dsubscribe_test.py264 infer_shape=False)
/external/tensorflow/tensorflow/python/grappler/
Dhierarchical_controller.py868 infer_shape=False,