Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for ReshapableTo1DTensor (0.19 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.td

      "llvm::ArrayRef<int>({" # !interleave(shape_2, ", ") # "}))">,
      "Checks if the given dimensions contain the same number of elements.">;
    
    def ReshapableTo1DTensor : Constraint<
      CPred<"quant::ReshapableTo1DTensor($0.getType().cast<ShapedType>())">,
      "Checks if the value dims are all ones except the right most dim">;
    
    def ReshapeTo1DTensor : NativeCodeCall<
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 03:24:59 UTC 2024
    - 8.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.cc

        }
        val2_result *= val2_shape.getDimSize(idx);
      }
    
      return val1_result == val2_result;
    }
    
    // Checks if a shape has dim sizes of all ones except the right most dim.
    bool ReshapableTo1DTensor(ShapedType rhs_shape) {
      for (auto rank = 0; rank < rhs_shape.getRank() - 1; rank++) {
        if (rhs_shape.getDimSize(rank) != 1) {
          return false;
        }
      }
      return true;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 13.3K bytes
    - Viewed (0)
Back to top