Home
last modified time | relevance | path

Searched defs:shape_dtype (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlower_static_tensor_list.cc199 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/
Dstateless_random_ops_test.py108 def wrap(op, dtype, shape, shape_dtype, seed, **kwargs): argument
139 def wrap(op, minval, maxval, shape, shape_dtype, dtype, seed, **kwargs): argument