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