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 | 191 IntegerType shape_dtype = rewriter->getIntegerType(32); in CreateSliceOpForTensorList() local
350 IntegerType shape_dtype = rewriter.getIntegerType(32); in matchAndRewriteImplWithSliceAndConcat() local
407 IntegerType shape_dtype = rewriter.getIntegerType(32); in matchAndRewriteImplWithDynamicUpdateSlice() local
476 Type shape_dtype = getElementTypeOrSelf(element_shape.getType()); in matchAndRewrite() local
636 Type shape_dtype = getElementTypeOrSelf(op.element_shape().getType()); in GetNumElements() local
728 IntegerType shape_dtype = rewriter.getIntegerType(32); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | stateless_random_ops_test.py | 104 def wrap(op, dtype, shape, shape_dtype, seed, **kwargs): argument
135 def wrap(op, minval, maxval, shape, shape_dtype, dtype, seed, **kwargs): argument
|