Searched defs:dyn_sizes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 94 ArrayRef<Value> dyn_sizes) { in GetInitTensor()
105 SmallVector<Value, 2> dyn_sizes(tensor_type.getRank()); local
259 SmallVector<Value, 2> dyn_sizes; in ExtractDynamicEinsumSizes() local
400 auto dyn_sizes = ExtractDynamicEinsumSizes( in matchAndRewrite() local
570 auto dyn_sizes = ExtractDynamicSizes(rewriter, loc, max_rank_arg); in matchAndRewrite() local
714 auto dyn_sizes = isLHLO ? SmallVector<Value, 2>() in matchAndRewrite() local
1235 auto dyn_sizes = in matchAndRewrite() local
1309 SmallVector<Value, 3> dyn_sizes; in matchAndRewrite() local
2085 SmallVector<Value, 8> dyn_sizes; in matchAndRewrite() local
2510 SmallVector<Value, 4> dyn_sizes; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tf_framework_legalize_to_llvm.cc | 541 SmallVector<Value, 4> dyn_sizes(result_type.getNumDynamicDims(), in matchAndRewrite() local
|