Searched refs:dyn_sizes (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ir/ |
| D | tf_framework_ops.td | 76 Variadic<Index>:$dyn_sizes,
89 "ValueRange":$dyn_sizes),
92 $_state.addOperands(dyn_sizes);
102 `(` $ctx (`,` $dyn_sizes^ )? `)` attr-dict `:` type($result)
|
| D | tf_framework_ops.cc | 72 unsigned dyn_sizes_count = op.dyn_sizes().size(); in Verify()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/tests/ |
| D | invalid.mlir | 4 // expected-error @+1 {{`dyn_sizes` count 1 does not match dynamic dimensions}}
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 89 ArrayRef<Value> dyn_sizes) { in GetInitTensor() argument
90 return b.create<linalg::InitTensorOp>(loc, dyn_sizes, type.getShape(), in GetInitTensor()
98 SmallVector<Value, 2> dyn_sizes; in ExtractDynamicSizes() local
101 dyn_sizes.push_back(b.create<DimOp>(loc, tensor, en.index())); in ExtractDynamicSizes()
103 return dyn_sizes; in ExtractDynamicSizes()
192 auto dyn_sizes = ExtractDynamicSizes(rewriter, loc, args[0]); in matchAndRewrite() local
194 GetInitTensor(rewriter, loc, result_type, dyn_sizes)); in matchAndRewrite()
384 auto dyn_sizes = isLHLO ? SmallVector<Value, 2>() in matchAndRewrite() local
393 : ValueRange{GetInitTensor(rewriter, loc, result_type, dyn_sizes)}, in matchAndRewrite()
851 auto dyn_sizes = isLHLO in matchAndRewrite() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | same_shape_propagation.cc | 227 fillShapeFromAllocLike(alloc.dyn_sizes(), type, shape); in build()
|
| D | tf_framework_legalize_to_llvm.cc | 85 llvm::to_vector<4>(transformed.dyn_sizes()), in matchAndRewrite()
|