- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for dynamic_slice (0.14 sec)
-
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) -
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)