Searched refs:size_0 (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/tests/ |
| D | ops.mlir | 9 %size_0 : index , %size_2 : index) {
11 %buf_1 = tf_framework.alloc(%ctx, %size_0, %size_2) : memref<?x10x?xi8>
17 %size_0 : index , %size_2 : index) {
18 %buf = tf_framework.alloc(%ctx, %size_0, %size_2) {
|
| D | embed_tf_framework.mlir | 9 func @tf_entry(%size_0 : index , %size_2 : index) -> index
11 %buf = alloc(%size_0, %size_2)[] : memref<?x10x?xf32>
13 std.return %size_0 : index
24 func @non_tf_entry(%size_0 : index , %size_2 : index) -> index {
25 std.return %size_0 : index
|
| D | tf_framework_legalize_to_llvm.mlir | 11 %size_0 : index , %size_2 : index) -> memref<?x10x?xf32> {
12 %buf = tf_framework.alloc(%ctx, %size_0, %size_2) : memref<?x10x?xf32>
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | reduce.cc | 434 const double size_0 = in BindArguments() local
436 const double size_1 = reduction_size / size_0; in BindArguments()
438 RETURN_IF_ERROR(args->SetFloat("inv_multiplier_2", 1.0 / size_0)); in BindArguments()
|
| /external/llvm-project/mlir/test/Conversion/StandardToLLVM/ |
| D | convert-dynamic-memref-ops.mlir | 431 %size_0 : index,
437 offset: [%offset], sizes: [%size_0, %size_1],
|