Searched refs:aliasing_output (Results 1 – 5 of 5) sorted by relevance
8 func @main(%arg0: tensor<1xf32> {tf.aliasing_output = 0 : i64}) -> (tensor<1xf32>) {
82 // CHECK-LABEL: func @main(%arg0: tensor<i1>, %arg1: tensor<f32> {tf.aliasing_output = 1 : i64, tf.…175 // CHECK-SAME: %{{[a-z0-9]+}}: tensor<f32> {tf.aliasing_output = 0 : i64}191 // CHECK-SAME: %{{[a-z0-9]+}}: tensor<f32> {tf.aliasing_output = 0 : i64}210 // CHECK-SAME: %{{[a-z0-9]+}}: tensor<f32> {tf.aliasing_output = 1 : i64}228 // CHECK-SAME: %{{[a-z0-9]+}}: tensor<f32> {tf.aliasing_output = 1 : i64}250 // CHECK-SAME: %[[ARG_0:[a-z0-9]+]]: tensor<f32> {tf.aliasing_output = 0 : i64}278 // CHECK-SAME: %{{[a-z0-9]+}}: tensor<f32> {tf.aliasing_output = 0 : i64}
24 // CHECK-SAME: func @main(%arg0: tensor<32xf32> {tf.aliasing_output = 0 : i64},69 …// CHECK-SAME: func @main(%arg0: tensor<32xf32>, %arg1: tensor<32xf32> {tf.aliasing_output = 0 : i…132 …// CHECK-SAME: func @main(%arg0: tensor<32xf32>, %arg1: tensor<32xf32> {tf.aliasing_output = 1 : i…
158 if (auto aliasing_output = main_func.getArgAttrOfType<mlir::IntegerAttr>( in GetOutputInfo() local160 output_to_input_alias[aliasing_output.getInt()] = i; in GetOutputInfo()
1415 auto aliasing_output = in RunOnFunction() local1417 if (!aliasing_output) continue; in RunOnFunction()1419 builder.SetUpAlias(/*output_index=*/{aliasing_output.getInt()}, in RunOnFunction()1422 builder.SetUpAlias(/*output_index=*/{aliasing_output.getInt()}, in RunOnFunction()