- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for in_shape (0.09 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
Value result = reduction.getResult(0); // The mean op needs to divide by the product of the reduced dimensions. if (std::is_same<OpTy, TF::MeanOp>::value) { Value in_shape = rewriter.create<shape::ShapeOfOp>(loc, op.getInput()); Value divisor_count = rewriter.create<arith::ConstantIndexOp>(loc, 1); for (size_t i = 0; i < input_shape.size(); ++i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// CHECK-DAG: [[RNG_SHAPE:%.*]] = mhlo.constant dense<4> : tensor<1xi64> // CHECK-DAG: [[RNG_LOWER:%.*]] = mhlo.constant dense<0> : tensor<i32> // CHECK-DAG: [[RNG_UPPER:%.*]] = mhlo.constant dense<4> : tensor<i32> // CHECK: [[SWAPS:%.*]] = "mhlo.rng"([[RNG_LOWER]], [[RNG_UPPER]], [[RNG_SHAPE]]) <{rng_distribution = #mhlo.rng_distribution<UNIFORM>}>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0)