- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for XlaReplicaId (0.18 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_patterns.td
) ), (replaceWithValue $output) ]>; //===----------------------------------------------------------------------===// // XlaReplicaId op. //===----------------------------------------------------------------------===// def : Pat<(TF_XlaReplicaIdOp), (TF_CastOp (MHLO_ReplicaIdOp), /*truncate=*/ConstBoolAttrFalse)>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 34.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
} func.func @replica_id() -> tensor<i32> { // CHECK: %[[ID:.*]] = mhlo.replica_id : tensor<ui32> // CHECK: %[[RESULT:.*]] = mhlo.convert %0 : (tensor<ui32>) -> tensor<i32> %0 = "tf.XlaReplicaId"() : () -> tensor<i32> func.return %0 : tensor<i32> } // CHECK: func @angle_c64 // CHECK-SAME: ([[ARG0:%.*]]: tensor<complex<f32>>) func.func @angle_c64(%arg0: tensor<complex<f32>>) -> tensor<f32> {
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/tensorflow/ir/tf_generated_ops.td
); let results = (outs TF_Tensor:$output ); TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>; } def TF_XlaReplicaIdOp : TF_Op<"XlaReplicaId", [Pure, TF_NoConstantFold]> { let summary = "Replica ID."; let arguments = (ins); let results = (outs TF_Int32Tensor:$id );
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)