- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 27 for reduce_op (2.24 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/reduce.h
reduce_op.getLoc(), reduce_op->getResult(0).getType(), operand, reduction_indices, /*keep_dim=*/rewriter.getBoolAttr(false)); auto tf_argreduce_op = rewriter.create<ArgReduce>( reduce_op.getLoc(), reduce_op->getResult(1).getType(), operand, reduction_indices); rewriter.replaceOp(reduce_op, {tf_reduce_op, tf_argreduce_op}); } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/reduce.cc
std::optional<bool> IsReduceOpLegal(mhlo::ReduceOp reduce_op) { if (succeeded(MatchReduceToArgMinMaxType1(reduce_op, true, true)) || succeeded(MatchReduceToArgMinMaxType1(reduce_op, false, true)) || succeeded(MatchReduceToArgMinMaxType1(reduce_op, true, false)) || succeeded(MatchReduceToArgMinMaxType1(reduce_op, false, false)) || succeeded(MatchReduceToArgMinMaxType2(reduce_op, false)) ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 20:53:17 UTC 2024 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize_layout.cc
getInvertPermutedAttr(reduce_op.getWindowDimensions()); DenseI64ArrayAttr new_window_strides = getInvertPermutedAttr(reduce_op.getWindowStrides()); DenseI64ArrayAttr new_base_dilations = getInvertPermutedAttr(reduce_op.getBaseDilations()); DenseI64ArrayAttr new_win_dilations = getInvertPermutedAttr(reduce_op.getWindowDilations());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 21:59:06 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-collective.mlir
// CHECK: "mhlo.all_reduce" // CHECK{LITERAL}: replica_groups = dense<[[0], [1]]> : tensor<2x1xi64> // CHECK-NOT: channel_handle // CHECK: mhlo.add %0 = "tf.XlaAllReduce"(%input, %group_assignment) {reduce_op = "Add", mode = "CrossReplica"} : (tensor<f32>, tensor<2x1xi32>) -> tensor<f32> func.return %0 : tensor<f32> } // ----- // CHECK: module attributes // CHECK-SAME{LITERAL}: tf2xla.collective_info.group_key = 0
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 15.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/util.h
if (!return_op) return failure(); if (return_op.getNumOperands() != 1) return failure(); ReductionOp reduce_op = dyn_cast_or_null<ReductionOp>( return_op.getOperands().front().getDefiningOp()); if (!reduce_op) return failure(); if (reduce_op.getLhs() != body.getArgument(0) || reduce_op.getRhs() != body.getArgument(1)) return failure(); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 08 11:35:25 UTC 2023 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/xla_broadcast.mlir
// CHECK-NEXT: "tf_device.cluster"() ({ // CHECK-NEXT: %[[GROUP:.*]] = "tf.Const"() // CHECK-SAME: [0, 1, 2, 3] // CHECK-NEXT: %[[REDUCED:.*]] = "tf.XlaAllReduce"(%[[ID]], %[[GROUP]]) <{mode = "CrossReplica", reduce_op = "Add"}> : (tensor<f32>, tensor<1x4xi32>) -> tensor<f32> // CHECK-NEXT: "tf.OpA"(%[[REDUCED]]) : (tensor<f32>) -> () tf_device.replicate {n = 4 : i32} { "tf_device.cluster"() ({
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
// This function tries to match that the "mhlo::ReduceOp" only has one // operand, one init_value and one result. LogicalResult MatchReduceOpOperand(mhlo::ReduceOp reduce_op) const { if (reduce_op.getInputs().size() != 1 || reduce_op.getInitValues().size() != 1 || reduce_op.getResults().size() != 1) return failure(); if (!mlir::isa<RankedTensorType>(reduce_op.getInputs()[0].getType()))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu-resource-read-for-write.mlir
func.func private @write_chain_func(%arg0: tensor<1xf32>) -> (tensor<1xf32>) { %cst = "tf.Const"() <{value = dense<[[0, 1]]> : tensor<1x2xi32>}> : () -> tensor<1x2xi32> %0 = "tf.XlaAllReduce"(%arg0, %cst) <{mode = "CrossReplica", reduce_op = "Add"}> : (tensor<1xf32>, tensor<1x2xi32>) -> tensor<1xf32> return %0 : tensor<1xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 16:54:40 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc
group_assignment_val)); StringAttr reduce_op = builder.getStringAttr("Add"); StringAttr mode = builder.getStringAttr("CrossReplica"); return builder.create<XlaAllReduceOp>(block_arg.getLoc(), block_arg.getType(), block_arg, group_assignment, reduce_op, mode); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0)