Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for dynamic_slice (0.14 sec)

  1. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

    // indices randomly generated via HLO rng_uniform ops. Otherwise, it is
    // translated into an HLO while op to first emulate shuffling indices using
    // HLO dynamic_slice and dynamic_update_slice ops, then finally HLO gather
    // with the shuffled indices.
    class ConvertRandomShuffleOp : public OpRewritePattern<TF::RandomShuffleOp> {
     public:
      using OpRewritePattern::OpRewritePattern;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

    }
    
    def TF_XlaDynamicSliceOp : TF_Op<"XlaDynamicSlice", [Pure, TF_NoConstantFold]> {
      let summary = "Wraps the XLA DynamicSlice operator, documented at";
    
      let description = [{
    https://www.tensorflow.org/performance/xla/operation_semantics#dynamicslice
    .
    
    DynamicSlice extracts a sub-array from the input array at dynamic
    start_indices. The size of the slice in each dimension is passed in
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top