Searched defs:infer_shape (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | tensor_array_ops.py | 59 infer_shape=True, argument
408 infer_shape=True, argument
671 infer_shape=True, argument
1027 infer_shape=True, argument
1358 infer_shape=True): argument
|
| H A D | map_fn.py | 50 infer_shape=True, argument
634 infer_shape=True, argument
|
| H A D | functional_ops.py | 445 infer_shape=True, argument
710 infer_shape=True, argument
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/shape_inference/ |
| H A D | infer_shape.py | 21 def infer_shape(gm, input_tensors): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_map_ops.py | 33 infer_shape=True, argument
|
| H A D | ragged_map_fn_op_test.py | 178 infer_shape=True, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/util/ |
| H A D | structure.py | 53 def _TensorArrayStructure(dtype, element_shape, dynamic_size, infer_shape): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/data_structures/ |
| H A D | tensor_array_ops_test.py | 61 def _make_ta(size, name, dtype=dtypes.float32, infer_shape=False): argument
|