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 | 146 IntegerType shape_dtype = rewriter->getIntegerType(32); in CreateSliceOpForTensorList() local
269 IntegerType shape_dtype = rewriter.getIntegerType(32); in matchAndRewrite() local
342 Type shape_dtype = getElementTypeOrSelf(element_shape.getType()); in matchAndRewrite() local
489 Type shape_dtype = getElementTypeOrSelf(op.element_shape().getType()); in GetNumElements() local
573 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, kwds, seed): argument
139 def wrap(op, minval, maxval, shape, shape_dtype, dtype, seed): argument
|