Home
last modified time | relevance | path

Searched refs:rng_bit_generator (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dxla_ops_test.py328 res1 = xla.rng_bit_generator(algorithm, k, shape, dtype=dtype)
329 res2 = xla.rng_bit_generator(algorithm, k, shape, dtype=dtype)
1054 res = xla.rng_bit_generator(algorithm, initial_state, shape, dtype=dtype)
1062 res = xla.rng_bit_generator(algorithm, initial_state, shape, dtype=dtype)
1070 res = xla.rng_bit_generator(algorithm, initial_state, shape, dtype=dtype)
1080 res = xla.rng_bit_generator(algorithm, initial_state, shape, dtype=dtype)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/python/
H A Dxla.py387 def rng_bit_generator(algorithm, initial_state, shape, dtype): function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/mhlo/
H A Dops.mlir1219 // CHECK-LABEL: func @rng_bit_generator
1220 func.func @rng_bit_generator(%arg0: tensor<2xui64>) -> (tensor<2xui64>, tensor<10x12xui32>) {
1224 …%2, %3 = "mhlo.rng_bit_generator"(%4) {rng_algorithm = #mhlo.rng_algorithm<DEFAULT>} : (tensor<2xu…
1230 func.func @rng_bit_generator(%arg0: tensor<2xui64>) -> (tensor<2xui64>, tensor<10x12xui32>) {
1235 …%2, %3 = "mhlo.rng_bit_generator"(%4) {rng_algorithm = #mhlo.rng_algorithm<DEFAULT>} : (tensor<2xu…
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/tests/
H A Dops_stablehlo.mlir1201 // CHECK-LABEL: func @rng_bit_generator
1202 func.func @rng_bit_generator(%arg0: tensor<2xui64>) -> (tensor<2xui64>, tensor<10x12xui32>) {
1206 …%2, %3 = "stablehlo.rng_bit_generator"(%4) {rng_algorithm = #stablehlo.rng_algorithm<DEFAULT>} : (…
1212 func.func @rng_bit_generator(%arg0: tensor<2xui64>) -> (tensor<2xui64>, tensor<10x12xui32>) {
1217 …%2, %3 = "stablehlo.rng_bit_generator"(%4) {rng_algorithm = #stablehlo.rng_algorithm<DEFAULT>} : (…
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
H A Dexport.mlir1551 …%0:2 = "mhlo.rng_bit_generator"(%arg) {rng_algorithm = #mhlo.rng_algorithm<PHILOX>} : (tensor<3xui…
H A Dimport.hlotxt1445 …// CHECK: %[[RNG0:.+]], %[[RNG1:.+]] = "mhlo.rng_bit_generator"(%[[ARG0]]) {rng_algorithm = #mhlo.…
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.td2304 def StableHLO_RngBitGeneratorOp : StableHLO_Op<"rng_bit_generator", [NoSideEffect]> {
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/
H A Dhlo_ops.td2529 def HLO_RngBitGeneratorOp : HLO_Op<"rng_bit_generator", [NoSideEffect]> {
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
H A Dlegalize-tf.mlir6482 …// CHECK-NEXT: %[[OUTPUT_STATE:.*]], %[[OUTPUT:.*]] = "mhlo.rng_bit_generator"(%[[STATE]]) {rng_al…