Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for Squeeze (0.08 sec)

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

        SmallVector<int64_t, 4> begin_indices(value_rank, 0);
        auto end_indices = llvm::to_vector<4>(value_type.getShape());
        SmallVector<int64_t, 4> strides(value_rank, 1);
    
        // All HLO slice+squeeze results used to replace the original tf.Unpack op.
        SmallVector<Value, 4> results;
        results.reserve(op.getNumResults());
    
        for (int i = 0, end = op.getNumResults(); i < end; ++i) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
Back to top