- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for RngReadAndSkip (0.17 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config_test.cc
func.func @main(%arg0: tensor<3xi64> {tf._user_specified_name = "resource", tf.aliasing_output = 3 : i64}) -> () attributes {tf.entry_function = {control_outputs = "stateful_normal/RngReadAndSkip,stateful_uniform/RngReadAndSkip,stateful_uniform_full_int/RngReadAndSkip", inputs = "stateful_normal_rngreadandskip_resource", outputs = "identity_RetVal,identity_1_RetVal,identity_2_RetVal"}} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 30 03:31:01 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter_test.cc
func.func @main(%arg0: tensor<3xi64> {tf._user_specified_name = "resource", tf.aliasing_output = 3 : i64}) -> () attributes {tf.entry_function = {control_outputs = "stateful_normal/RngReadAndSkip,stateful_uniform/RngReadAndSkip,stateful_uniform_full_int/RngReadAndSkip", inputs = "stateful_normal_rngreadandskip_resource", outputs = "identity_RetVal,identity_1_RetVal,identity_2_RetVal"}} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/decompose_resource_ops.mlir
} // Test that tf.RngReadAndSkip op is decomposed. // CHECK-LABEL: func @decompose_rng_read_and_skip_op func.func @decompose_rng_read_and_skip_op(%resource: tensor<!tf_type.resource<tensor<3xi64>>>) -> tensor<3xi64> { %0 = "tf_device.cluster"() ({ // We rely on the TensorFlow StatefulRandomOpsTest to check it is lowered // correctly. // CHECK-NOT: tf.RngReadAndSkip
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 51.3K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
"ResourceScatterMul", "ResourceScatterNdAdd", "ResourceScatterNdSub", "ResourceScatterNdUpdate", "ResourceScatterSub", "ResourceScatterUpdate", "RngReadAndSkip", "RngSkip", "Roll", "ScatterNd", "SegmentSumV2", "SegmentProdV2", "SegmentMinV2", "SegmentMaxV2", "SelfAdjointEigV2",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
); let results = (outs TF_FloatTensor:$product ); TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>; } def TF_RngReadAndSkipOp : TF_Op<"RngReadAndSkip", []> { let summary = "Advance the counter of a counter-based RNG."; let description = [{ The state of the RNG after `rng_read_and_skip(n)` will be the same as that after `uniform([n])`
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)