- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for rng_bit_generator (0.32 sec)
-
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/vhlo.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 14 19:15:40 UTC 2024 - 31.9K bytes - Viewed (1) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// CHECK-NEXT: %1 = mhlo.constant dense<3> : tensor<i32> %cst_0 = "tf.Const"() {value = dense<3> : tensor<i32>} : () -> tensor<i32> // CHECK-NEXT: %[[OUTPUT_STATE:.*]], %[[OUTPUT:.*]] = "mhlo.rng_bit_generator"(%[[STATE]]) <{rng_algorithm = #mhlo.rng_algorithm<DEFAULT>}> : (tensor<2xui64>) -> (tensor<2xui64>, tensor<10x12xui32>) // CHECK-NEXT: return %[[OUTPUT_STATE]], %[[OUTPUT]] : tensor<2xui64>, tensor<10x12xui32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
} else if (alg == tensorflow::RNG_ALG_AUTO_SELECT) { return xla::RandomAlgorithm::RNG_DEFAULT; } return std::nullopt; } // Converts tf.XlaRngBitGenerator op to mhlo.RngBitGenerator op. class ConvertXlaRngBitGeneratorOp : public OpRewritePattern<TF::XlaRngBitGeneratorOp> { public: using OpRewritePattern::OpRewritePattern;
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/tensorflow/ir/tf_generated_ops.td
def TF_XlaRngBitGeneratorOp : TF_Op<"XlaRngBitGenerator", [Pure]> { let summary = "Stateless PRNG bit generator."; let description = [{ Wraps the XLA RngBitGenerator operator, documented at https://www.tensorflow.org/performance/xla/operation_semantics#rngbitgenerator. }]; let arguments = (ins Arg<TF_Int32Tensor, [{The PRNG algorithm to use, one of
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)