Searched defs:shape_dtype (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 199 IntegerType shape_dtype = rewriter->getIntegerType(32); in CreateSliceOpForTensorList() local
338 IntegerType shape_dtype = rewriter.getIntegerType(32); in matchAndRewrite() local
414 Type shape_dtype = getElementTypeOrSelf(element_shape.getType()); in matchAndRewrite() local
564 Type shape_dtype = getElementTypeOrSelf(op.element_shape().getType()); in GetNumElements() local
652 IntegerType shape_dtype = rewriter.getIntegerType(32); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | stateless_random_ops_test.py | 108 def wrap(op, dtype, shape, shape_dtype, seed, **kwargs): argument
139 def wrap(op, minval, maxval, shape, shape_dtype, dtype, seed, **kwargs): argument
|