- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for CollectiveReduce (0.26 sec)
-
tensorflow/c/eager/c_api_test_util.cc
return op; } TFE_Op* AllReduceOp(TFE_Context* ctx, TFE_TensorHandle* in, int group_size) { TF_Status* status = TF_NewStatus(); TFE_Op* op = TFE_NewOp(ctx, "CollectiveReduce", status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_OpAddInput(op, in, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TF_DeleteStatus(status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_op_base.td
def TF_WriteTrainingPredictions : MemoryEffects<[MemWrite<TF_WriteTrainingPredictionsResource>]>; def TF_RandomGeneratorSideEffect : MemoryEffects<[MemWrite<TF_RandomGeneratorResource>]>; // Special effect for keeping `CollectiveReduce` ops in order. def TF_CollectiveReduceOrderingEffect : MemoryEffects<[MemWrite<TF_CollectiveReduceOrderingResource>]>; // Special effect for keeping `NcclAllReduce` ops on the same device in order.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 30.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
let results = (outs Res<TF_NumberTensor, [{The permuted input.}]>:$output ); TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>; } def TF_CollectiveReduceOp : TF_Op<"CollectiveReduce", [TF_SameOperandsAndResultTypeResolveRef]> { let summary = [{ Mutually reduces multiple tensors of identical type and shape. }]; let arguments = (ins TF_FpOrI32OrI64Tensor:$input,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)