- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for rng_bit_generator_v1 (0.27 sec)
-
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/vhlo.mlir
// CHECK-NEXT: return %0 : tensor<2x3x2xi32> // CHECK-NEXT:} func.func @rng_bit_generator(%arg0: tensor<2xui64>) -> (tensor<2xui64>, tensor<10x12xui32>) { %output_state, %output = "vhlo.rng_bit_generator_v1"(%arg0) <{rng_algorithm = #vhlo<rng_algorithm_v1 DEFAULT>}> : (tensor<2xui64>) -> (tensor<2xui64>, tensor<10x12xui32>) func.return %output_state, %output : tensor<2xui64>, tensor<10x12xui32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 14 19:15:40 UTC 2024 - 31.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
rewriter.getContext(), *mlir::mhlo::symbolizeRngAlgorithm(xla_alg.value())); auto rng_bit_generator_op = rewriter.create<mhlo::RngBitGeneratorOp>( loc, op.getResultTypes(), algorithm_attr, op.getInitialState()); rewriter.replaceOp(op, rng_bit_generator_op.getResults()); return success(); } }; // Converts tf.XlaVariadicReduceV2 to mhlo.Reduce
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)