- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for XlaVariadicReduceV2 (0.31 sec)
-
tensorflow/compiler/jit/xla_ops_on_regular_devices.cc
REGISTER_KERNEL_BUILDER(Name("XlaVariadicReduce").Device(DEVICE), \ XlaCompileOnDemandOp); \ REGISTER_KERNEL_BUILDER(Name("XlaVariadicReduceV2").Device(DEVICE), \ XlaCompileOnDemandOp); \ REGISTER_KERNEL_BUILDER(Name("XlaReducePrecision").Device(DEVICE), \
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 19 19:55:14 UTC 2022 - 8.8K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.cc
"XlaSpmdShardToFullShape", "XlaSvd", "XlaVariadicReduceV2", "XlaVariadicSort", "XlaWhile"}; static bool NodeCanTriggerXlaCompilation(const NodeDef& node) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir
} // CHECK-LABEL: testXlaReduceToXlaVariadicReduceV2 func.func @testXlaReduceToXlaVariadicReduceV2(%arg0: tensor<*xbf16>, %arg1: tensor<*xbf16>) -> tensor<*xbf16> { // CHECK: "tf.XlaVariadicReduceV2"(%arg0, %arg1) <{dimensions_to_reduce = [], operandSegmentSizes = array<i32: 1, 1>, reducer = @sum1}> {device = "/job:localhost/replica:0/task:0/device:GPU:0"} : (tensor<*xbf16>, tensor<*xbf16>) -> tensor<*xbf16>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 132.1K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
"XlaSharding", "XlaSort", "XlaSplitND", "XlaSpmdFullToShardShape", "XlaSpmdShardToFullShape", "XlaSvd", "XlaVariadicReduce", "XlaVariadicReduceV2", "XlaVariadicSort", "XlaWhile", "Zeta", "_Arg", "_ArrayToList", "_ListToArray", "_Retval"}; return result; }
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_ops_n_z.cc
} } if (op.getDimensionsToReduce().size() > input_ty_0.getRank()) { return op.emitOpError() << "Invalid dimensions_to_reduce argument to XlaVariadicReduceV2"; } } for (int i = 0; i < n_inputs; ++i) { auto init_value_ty_i = init_values_ty[i].cast<ShapedType>(); if (init_value_ty_i.hasRank() && init_value_ty_i.getRank() != 0) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
return ::xla::ConvertPrecisionConfig(&precision, builder); } //===----------------------------------------------------------------------===// // XlaVariadicReduceV2 op utilities. //===----------------------------------------------------------------------===// static void BuildBodyWithCall(PatternRewriter &rewriter, const Location &loc,
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/tf2xla/tests/legalize-tf.mlir
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 description = [{ Semantics are documented at https://www.tensorflow.org/performance/xla/operation_semantics#variadic_reduce. This version is limited to operands of the same dtype. XlaVariadicReduceV2 is a version that supports heterogeneous operands. }]; let arguments = (ins
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)