- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for UnsortedSegmentProdOp (0.19 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc
DimensionVector out_dimensions_; }; // Calculates the flattened shapes for dynamic shaped operands in // mhlo.dot_general: // 1. flattened_out_dim = UnsortedSegmentProdOp(operand_shape, out_axes) // 2. flattened_contracting_dim = UnsortedSegmentProdOp(operand_shape, // contracting_axes) // 3. batch_dimensions = Gather(operand_shape, batch_axes) // 4. flattened_shape = Concat(batch_dimensions, flattened_out_dim,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
// Build UnsortedSegmentProdOp Type segProdresultType = RankedTensorType::get(num_reshape_segids, rewriter->getIntegerType(32)); auto segids_tensor = createI32ConstantOp(reshape_segids, loc, rewriter); auto num_reshape_segids_tensor = createI32ConstOp(num_reshape_segids, loc, rewriter); auto segprod = rewriter->create<TF::UnsortedSegmentProdOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc
TypeID::get<TF::TopKV2Op>(), TypeID::get<TF::_UnaryOpsCompositionOp>(), TypeID::get<TF::UnsortedSegmentMaxOp>(), TypeID::get<TF::UnsortedSegmentMinOp>(), TypeID::get<TF::UnsortedSegmentProdOp>(), TypeID::get<TF::UnsortedSegmentSumOp>(), TypeID::get<TF::XdivyOp>(), TypeID::get<TF::XlaSendTPUEmbeddingGradientsOp>(), TypeID::get<TF::XlaAllReduceOp>(), TypeID::get<TF::XlaGatherOp>(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 21.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
DimensionVector out_dimensions_; }; // Calculates the flattened shapes for dynamic shaped operands in // mhlo.dot_general: // 1. flattened_out_dim = UnsortedSegmentProdOp(operand_shape, out_axes) // 2. flattened_contracting_dim = UnsortedSegmentProdOp(operand_shape, // contracting_axes) // 3. batch_dimensions = Gather(operand_shape, batch_axes) // 4. flattened_shape = Concat(batch_dimensions, flattened_out_dim,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
LogicalResult UnsortedSegmentMaxOp::verify() { return VerifyUnsortedSegmentReduction(*this); } LogicalResult UnsortedSegmentMinOp::verify() { return VerifyUnsortedSegmentReduction(*this); } LogicalResult UnsortedSegmentProdOp::verify() { return VerifyUnsortedSegmentReduction(*this); } LogicalResult UnsortedSegmentSumOp::verify() { return VerifyUnsortedSegmentReduction(*this); }
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
rewriter); } }; class ConvertUnsortedSegmentProdOp : public GenericConvertUnsortedSegmentReductionOp< ConvertUnsortedSegmentProdOp, TF::UnsortedSegmentProdOp, MulOp> { public: using GenericConvertUnsortedSegmentReductionOp:: GenericConvertUnsortedSegmentReductionOp; static Value GetInitialValue(Type reduce_element_type, 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)